Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Leavesfly/TinyDL-0.01
/ functions
Functions
1,708 in github.com/Leavesfly/TinyDL-0.01
⨍
Functions
1,708
◇
Types & classes
216
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2OutputHead.java:49
↓ 1 callers
Method
init
初始化仿射层的参数 使用Xavier初始化方法初始化权重矩阵
src/main/java/io/leavesfly/tinydl/nnet/layer/dnn/AffineLayer.java:57
↓ 1 callers
Method
init
初始化RNN层的参数 包括输入到隐藏状态权重、隐藏状态到隐藏状态权重和偏置项
src/main/java/io/leavesfly/tinydl/nnet/layer/rnn/SimpleRnnLayer.java:113
↓ 1 callers
Method
init
初始化GRU层的参数 包括更新门、重置门和候选状态的权重矩阵及偏置项
src/main/java/io/leavesfly/tinydl/nnet/layer/rnn/GruLayer.java:197
↓ 1 callers
Method
init
初始化模型 <p> 初始化编码器和解码器的参数。该方法在模型第一次使用前会自动调用。
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/EncoderDecoder.java:120
↓ 1 callers
Method
init
初始化编码器的所有层 该方法会创建并配置所有的子层,包括嵌入层、LSTM层和Dropout层。
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqEncoder.java:136
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/TransformerDecoder.java:66
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/TransformerEncoder.java:65
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/GPT2Block.java:62
↓ 1 callers
Method
initLayers
初始化网络层
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:159
↓ 1 callers
Method
initState
(NdArray encoderOutput)
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqDecoder.java:76
↓ 1 callers
Method
initState
初始化解码器的内部状态 该方法必须在执行layerForward之前调用,用于接收编码器的输出状态。 不同类型的解码器可能需要不同的初始化策略。 @param encoderOutput 编码器的输出状态,不能为null @throws IllegalArgumentException 当enc
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Decoder.java:74
↓ 1 callers
Method
initializeBias
初始化偏置参数
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:170
↓ 1 callers
Method
initializeModelInfo
初始化模型信息 包括输入输出形状、参数数量、架构类型等基本信息
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:59
↓ 1 callers
Method
initializeRandomParameters
初始化随机参数 @param model 要初始化参数的模型
src/main/java/io/leavesfly/tinydl/example/ModelSerializationExample.java:97
↓ 1 callers
Method
initializeWeights
初始化权重参数(使用Xavier初始化)
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:143
↓ 1 callers
Method
input2NdArray
将输入数据转换为NdArray @param input 输入数据 @return NdArray表示
src/main/java/io/leavesfly/tinydl/mlearning/inference/Translator.java:24
↓ 1 callers
Method
isLayersInitialized
()
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqDecoder.java:136
↓ 1 callers
Method
isLayersInitialized
检查层是否已初始化
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqEncoder.java:309
↓ 1 callers
Method
isParallelTrainingEnabled
获取并行训练状态 @return true 如果并行训练已启用
src/main/java/io/leavesfly/tinydl/mlearning/Trainer.java:379
↓ 1 callers
Method
isStateInitialized
检查解码器状态是否已初始化 @return 如果状态已初始化则返回true,否则返回false
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Decoder.java:81
↓ 1 callers
Method
isSuccess
()
src/main/java/io/leavesfly/tinydl/mlearning/parallel/ParallelBatchProcessor.java:116
↓ 1 callers
Method
isTerminated
检查是否到达终止状态 @return 是否终止
src/main/java/io/leavesfly/tinydl/modality/rl/environment/CartPoleEnvironment.java:153
↓ 1 callers
Method
isUsingBaseline
是否使用基线 @return 是否使用基线
src/main/java/io/leavesfly/tinydl/modality/rl/agent/REINFORCEAgent.java:441
↓ 1 callers
Method
isValidAction
(Variable action)
src/main/java/io/leavesfly/tinydl/modality/rl/environment/CartPoleEnvironment.java:173
↓ 1 callers
Method
isValidAction
(Variable action)
src/main/java/io/leavesfly/tinydl/modality/rl/environment/GridWorldEnvironment.java:234
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:281
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoELayer.java:166
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoEExpertNetwork.java:140
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoEGatingNetwork.java:87
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/modality/nlp/block/MoETransformerBlock.java:174
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/LayerNorm.java:65
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/TransformerDecoderLayer.java:76
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/FeedForward.java:68
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2TokenEmbedding.java:80
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2OutputHead.java:69
↓ 1 callers
Method
layerForward
层的前向传播计算 根据输入的词汇索引从权重矩阵中查找对应的词向量 @param inputs 输入变量数组,包含词汇索引 @return 前向传播结果变量
src/main/java/io/leavesfly/tinydl/nnet/layer/embedd/Embedding.java:76
↓ 1 callers
Method
layerForward
ReLU激活函数的前向传播方法 @param inputs 输入变量数组,通常只包含一个输入变量 @return 经过ReLU激活函数处理后的输出变量
src/main/java/io/leavesfly/tinydl/nnet/layer/activate/ReLuLayer.java:57
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqDecoder.java:87
↓ 1 callers
Method
layerForward
执行前向传播 <p> 该方法实现了完整的序列到序列转换流程。 @param inputs 输入参数,期望包含两个参数: inputs[0] - 编码器输入序列(源序列) inputs[1] - 解码器输入序列(目标序列的前缀) @return
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/EncoderDecoder.java:153
↓ 1 callers
Method
layerForward
执行编码器的前向传播 该方法实现了完整的编码流程:词嵌入 -> LSTM处理 -> Dropout正则化。 @param inputs 输入参数,期望包含一个参数: inputs[0] - 输入序列,形状为 [batch_size, seq_length] @retu
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqEncoder.java:190
↓ 1 callers
Method
layerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/GPT2Block.java:70
↓ 1 callers
Method
learn
(Experience experience)
src/main/java/io/leavesfly/tinydl/modality/rl/agent/REINFORCEAgent.java:232
↓ 1 callers
Method
learnBatch
(Experience[] experiences)
src/main/java/io/leavesfly/tinydl/modality/rl/agent/DQNAgent.java:178
↓ 1 callers
Method
loadCheckpoint
加载训练检查点 @param filePath 检查点文件路径 @return 检查点信息(包含模型、轮次、损失等)
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:223
↓ 1 callers
Method
loadParameters
加载模型参数到现有模型中 @param model 目标模型 @param filePath 参数文件路径
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:147
↓ 1 callers
Method
loadParameters
加载参数到当前模型 @param filePath 参数文件路径
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:171
↓ 1 callers
Method
lr
计算调整后的学习率 @return 调整后的学习率
src/main/java/io/leavesfly/tinydl/mlearning/optimize/Adam.java:97
↓ 1 callers
Method
lt
小于比较运算,比较当前数组元素是否小于另一个数组对应元素 @param other 另一个操作数数组 @return 比较结果数组,1.0表示小于,0.0表示不小于 @throws IllegalArgumentException 当两个数组形状不一致时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:598
↓ 1 callers
Method
matmul2D
执行二维矩阵乘法
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2OutputHead.java:114
↓ 1 callers
Method
meanSquaError
均方误差损失函数 @param y 真实值 @param x 预测值 @return 均方误差
src/main/java/io/leavesfly/tinydl/example/regress/LineExam.java:103
↓ 1 callers
Method
meanSquaredError
均方误差损失 计算当前变量与目标变量之间的均方误差损失 @param other 目标变量 @return 均方误差损失值
src/main/java/io/leavesfly/tinydl/func/Variable.java:638
↓ 1 callers
Method
mergeAlongAxis
沿指定轴合并多个数组(通用方法) @param result 结果数组 @param axis 合并轴 @param ndArrays 待合并数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArrayUtil.java:117
↓ 1 callers
Method
mergeMatrixAxis1
优化的矩阵按列合并方法 @param result 结果数组 @param ndArrays 待合并数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArrayUtil.java:94
↓ 1 callers
Method
ndArray2Output
将NdArray转换为输出数据 @param ndArray NdArray对象 @return 输出数据
src/main/java/io/leavesfly/tinydl/mlearning/inference/Translator.java:31
↓ 1 callers
Method
numericalDiff
数字微分函数的通用求导 使用数值微分方法计算函数在指定点的导数 @param func 要求导的函数 @param input 输入值 @param eps 微小增量,默认为0.00001 @return 导数结果
src/main/java/io/leavesfly/tinydl/utils/Util.java:25
↓ 1 callers
Method
parallelTrain
并行训练实现 将batch分配给多个线程并行处理,然后聚合梯度并更新参数 @param shuffleData 是否打乱数据
src/main/java/io/leavesfly/tinydl/mlearning/Trainer.java:190
↓ 1 callers
Method
performIm2Col
执行Im2Col操作 @param input 输入数据 @return 列格式的输入数据
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:220
↓ 1 callers
Method
performMatrixMultiplication
执行矩阵乘法操作 @return 矩阵乘法结果
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:241
↓ 1 callers
Method
performanceAnalysis
性能分析
src/main/java/io/leavesfly/tinydl/example/nlp/MoEGPTExample.java:428
↓ 1 callers
Method
pointwiseConvolution
逐点卷积实现
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/DepthwiseSeparableConvLayer.java:146
↓ 1 callers
Method
predictNextToken
预测下一个token @param tokenIds 输入token序列 @return 最可能的下一个token ID
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:271
↓ 1 callers
Method
prepareBatch
准备训练批次
src/main/java/io/leavesfly/tinydl/example/nlp/MoEGPTExample.java:229
↓ 1 callers
Method
prepareTrainingData
准备训练数据
src/main/java/io/leavesfly/tinydl/example/nlp/MoEGPTExample.java:137
↓ 1 callers
Method
preprocessTexts
预处理文本数据 将原始文本转换为token序列,并进行填充或截断处理
src/main/java/io/leavesfly/tinydl/mlearning/dataset/GPT2TextDataset.java:64
↓ 1 callers
Method
printAllExpertStatistics
打印所有层的专家使用统计
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:383
↓ 1 callers
Method
printExpertStatistics
打印专家使用统计信息
src/main/java/io/leavesfly/tinydl/modality/nlp/block/MoETransformerBlock.java:268
↓ 1 callers
Method
printExpertUsageStatistics
打印专家使用统计信息
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoELayer.java:318
↓ 1 callers
Method
printModelInfo
打印模型信息
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:477
↓ 1 callers
Method
printModelInfo
打印模型信息
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:326
↓ 1 callers
Method
printSimpleExpertUsage
简化的专家使用统计打印
src/main/java/io/leavesfly/tinydl/example/nlp/MoEGPTExample.java:293
↓ 1 callers
Method
printStatus
打印智能体当前状态
src/main/java/io/leavesfly/tinydl/modality/rl/agent/BanditAgent.java:260
↓ 1 callers
Method
printVocabInfo
打印词汇表信息
src/main/java/io/leavesfly/tinydl/modality/nlp/SimpleTokenizer.java:345
↓ 1 callers
Method
processBatchesInParallel
并行处理批次数据 @param batches 批次列表 @param epoch 当前轮次
src/main/java/io/leavesfly/tinydl/mlearning/Trainer.java:233
↓ 1 callers
Method
processBatchesSequentially
顺序处理批次数据(备用方案) @param batches 批次列表 @param epoch 当前轮次
src/main/java/io/leavesfly/tinydl/mlearning/Trainer.java:314
↓ 1 callers
Method
putAll
递归收集所有参数 @param allParams 用于存储所有参数的Map
src/main/java/io/leavesfly/tinydl/nnet/Block.java:103
↓ 1 callers
Method
relu
ReLU运算 对变量执行ReLU运算,将负值置为0 @return ReLU运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:468
↓ 1 callers
Method
renderASCII
ASCII艺术渲染
src/main/java/io/leavesfly/tinydl/modality/rl/environment/CartPoleEnvironment.java:192
↓ 1 callers
Method
reset
重置环境到初始状态 @return 初始状态
src/main/java/io/leavesfly/tinydl/modality/rl/Environment.java:71
↓ 1 callers
Method
resetAllExpertStatistics
重置所有MoE块的专家使用统计
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:373
↓ 1 callers
Method
resetExpertStatistics
重置MoE专家使用统计
src/main/java/io/leavesfly/tinydl/modality/nlp/block/MoETransformerBlock.java:261
↓ 1 callers
Method
resetState
重置Block中所有RNN层的状态
src/main/java/io/leavesfly/tinydl/nnet/Block.java:113
↓ 1 callers
Method
resetState
重置LSTM层的内部状态 在处理新序列之前应调用此方法
src/main/java/io/leavesfly/tinydl/nnet/layer/rnn/LstmLayer.java:162
↓ 1 callers
Method
resetUsageStatistics
重置专家使用统计
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoELayer.java:291
↓ 1 callers
Method
reshapeFilterWeights
重塑滤波器权重以进行矩阵乘法 @return 重塑后的权重
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:231
↓ 1 callers
Method
reshapeFrom2D
将二维矩阵重塑回三维张量
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2OutputHead.java:106
↓ 1 callers
Method
reshapeOutput
重塑输出为正确的形状 @param output 矩阵乘法的输出 @param batchSize 批处理大小 @return 重塑后的输出
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:252
↓ 1 callers
Method
reshapeTo2D
将三维张量重塑为二维矩阵
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2OutputHead.java:98
↓ 1 callers
Method
resumeFromCheckpoint
从检查点恢复训练 @param filePath 检查点文件路径 @return 恢复的模型
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:245
↓ 1 callers
Method
runComparisonExperiment
运行算法比较实验
src/main/java/io/leavesfly/tinydl/example/rl/MultiArmedBanditExample.java:69
↓ 1 callers
Method
runDetailedExperiment
运行详细实验(显示中间过程)
src/main/java/io/leavesfly/tinydl/example/rl/MultiArmedBanditExample.java:186
↓ 1 callers
Method
runDetailedSingleExperiment
运行单次详细实验
src/main/java/io/leavesfly/tinydl/example/rl/MultiArmedBanditExample.java:95
↓ 1 callers
Method
runSingleExperiment
运行单次实验
src/main/java/io/leavesfly/tinydl/example/rl/MultiArmedBanditExample.java:126
↓ 1 callers
Method
sample
随机采样一批经验 @param batchSize 批次大小 @return 采样的经验数组
src/main/java/io/leavesfly/tinydl/modality/rl/ReplayBuffer.java:73
↓ 1 callers
Method
sampleFromProbabilities
从概率分布中采样动作 @param probabilities 概率分布 @return 采样的动作
src/main/java/io/leavesfly/tinydl/modality/rl/agent/REINFORCEAgent.java:182
↓ 1 callers
Method
saveCheckpoint
保存训练检查点(包含模型状态和训练信息) @param model 模型 @param epoch 当前训练轮次 @param loss 当前损失 @param filePath 保存路径
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:197
↓ 1 callers
Method
saveModelCompressed
保存压缩模型 @param filePath 保存路径
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:115
↓ 1 callers
Method
saveParameterStats
保存参数统计信息到文本文件 @param parameters 参数映射 @param filePath 文件路径
src/main/java/io/leavesfly/tinydl/mlearning/ParameterManager.java:288
↓ 1 callers
Method
saveParameters
仅保存模型参数 @param model 模型 @param filePath 保存路径
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:126
↓ 1 callers
Method
selectArm
()
src/main/java/io/leavesfly/tinydl/modality/rl/agent/ThompsonSamplingBanditAgent.java:107
← previous
next →
601–700 of 1,708, ranked by callers