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
forward
前向传播计算均方误差 计算公式:MSE = Σ(predict - label)² / n 其中n为样本数量 @param inputs 输入的NdArray数组,包含预测值和真实标签 @return 均方误差值
src/main/java/io/leavesfly/tinydl/func/loss/MeanSE.java:25
↓ 1 callers
Method
forward
前向传播计算Sigmoid交叉熵损失 计算公式:Loss = -[y log(σ(x)) + (1-y) log(1-σ(x))] 其中σ(x)为Sigmoid函数,y为真实标签 @param inputs 输入的NdArray数组,包含预测值和真实标签 @return Sigmoid交叉熵损失
src/main/java/io/leavesfly/tinydl/func/loss/SigmoidCE.java:30
↓ 1 callers
Method
forward
前向传播计算幂函数 计算输入值的幂次方:x^pow @param inputs 输入的NdArray数组,长度为1 @return 幂函数值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Pow.java:35
↓ 1 callers
Method
forward
前向传播计算Tanh 计算Tanh函数值:(e^x - e^(-x)) / (e^x + e^(-x)) @param inputs 输入的NdArray数组,长度为1 @return Tanh函数值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Tanh.java:23
↓ 1 callers
Method
forward
前向传播计算平方 计算输入值的平方:x² @param inputs 输入的NdArray数组,长度为1 @return 平方值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Squ.java:24
↓ 1 callers
Method
forward
前向传播计算Sigmoid 计算Sigmoid函数值:1 / (1 + e^(-x)) @param inputs 输入的NdArray数组,长度为1 @return Sigmoid函数值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Sigmoid.java:24
↓ 1 callers
Method
forward
前向传播计算余弦 计算输入值的余弦值:cos(x) @param inputs 输入的NdArray数组,长度为1 @return 余弦值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Cos.java:24
↓ 1 callers
Method
forward
前向传播计算裁剪 将输入值限制在[min, max]范围内。 @param inputs 输入的NdArray数组,长度为1 @return 裁剪后的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Clip.java:37
↓ 1 callers
Method
forward
前向传播计算最小值 计算输入数组沿指定轴的最小值。 @param inputs 输入的NdArray数组,长度为1 @return 最小值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Min.java:31
↓ 1 callers
Method
forward
前向传播计算正弦 计算输入值的正弦值:sin(x) @param inputs 输入的NdArray数组,长度为1 @return 正弦值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Sin.java:23
↓ 1 callers
Method
forward2D
2D输入的前向传播 (N, C)
src/main/java/io/leavesfly/tinydl/nnet/layer/norm/BatchNorm.java:173
↓ 1 callers
Method
forward4D
4D输入的前向传播 (N, C, H, W)
src/main/java/io/leavesfly/tinydl/nnet/layer/norm/BatchNorm.java:120
↓ 1 callers
Method
forwardAdaptiveAveragePooling
自适应平均池化前向传播
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/PoolingLayer.java:198
↓ 1 callers
Method
forwardAdaptiveMaxPooling
自适应最大池化前向传播
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/PoolingLayer.java:158
↓ 1 callers
Method
forwardAveragePooling
平均池化前向传播
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/PoolingLayer.java:141
↓ 1 callers
Method
forwardCBOW
CBOW前向传播 @param input 输入变量 @return 前向传播结果
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:301
↓ 1 callers
Method
forwardMaxPooling
最大池化前向传播
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/PoolingLayer.java:123
↓ 1 callers
Method
forwardSkipGram
Skip-gram前向传播 @param input 输入变量 @return 前向传播结果
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:281
↓ 1 callers
Method
generate
生成文本的前向传播(用于推理) @param tokenIds 输入token序列 @return 下一个token的概率分布
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:321
↓ 1 callers
Method
generate
生成文本的前向传播(用于推理) @param tokenIds 输入token序列 @return 下一个token的概率分布
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:261
↓ 1 callers
Method
generateCBOWSamples
生成CBOW训练样本 CBOW: 给定上下文词,预测中心词
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/Word2VecDataSet.java:170
↓ 1 callers
Method
generateCacheKey
生成缓存键
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/Col2ImUtil.java:19
↓ 1 callers
Method
generateCacheKey
生成缓存键
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/Im2ColUtil.java:20
↓ 1 callers
Method
generateFromPrompt
从提示生成文本 @param gpt2Model GPT-2模型实例 @param tokenizer 分词器实例 @param prompt 提示文本 @param maxNewTokens 最大新生成token数 @return 生成的文本
src/main/java/io/leavesfly/tinydl/example/nlp/GPT2Example.java:262
↓ 1 callers
Method
generatePerformanceSummary
生成性能分析总结
src/test/java/io/leavesfly/tinydl/test/cnn/CnnPerformanceBenchmark.java:288
↓ 1 callers
Method
generateReward
根据指定臂的奖励分布生成奖励 @param armIndex 臂的索引 @return 生成的奖励值
src/main/java/io/leavesfly/tinydl/modality/rl/environment/MultiArmedBanditEnvironment.java:151
↓ 1 callers
Method
generateSkipGramSamples
生成Skip-gram训练样本 Skip-gram: 给定中心词,预测上下文词
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/Word2VecDataSet.java:146
↓ 1 callers
Method
generateText
生成文本 @param gpt2Model GPT-2模型实例 @param tokenizer 分词器实例
src/main/java/io/leavesfly/tinydl/example/nlp/GPT2Example.java:230
↓ 1 callers
Method
generateText
简化的文本生成函数
src/main/java/io/leavesfly/tinydl/example/nlp/MoEGPTExample.java:366
↓ 1 callers
Method
getActionDim
获取动作空间维度 @return 动作空间维度
src/main/java/io/leavesfly/tinydl/modality/rl/Environment.java:102
↓ 1 callers
Method
getActionIndex
从动作变量中获取动作索引 @param action 动作变量 @return 动作索引
src/main/java/io/leavesfly/tinydl/modality/rl/policy/EpsilonGreedyPolicy.java:162
↓ 1 callers
Method
getActionName
获取动作名称 @param action 动作值 @return 动作名称
src/main/java/io/leavesfly/tinydl/modality/rl/environment/GridWorldEnvironment.java:272
↓ 1 callers
Method
getActionProbabilities
(Variable state)
src/main/java/io/leavesfly/tinydl/modality/rl/policy/EpsilonGreedyPolicy.java:63
↓ 1 callers
Method
getActionProbability
(Variable state, Variable action)
src/main/java/io/leavesfly/tinydl/modality/rl/policy/EpsilonGreedyPolicy.java:87
↓ 1 callers
Method
getAverageGradients
等待所有梯度收集完成并返回平均梯度 @return 平均后的梯度映射 @throws InterruptedException 如果等待被中断
src/main/java/io/leavesfly/tinydl/mlearning/parallel/GradientAggregator.java:76
↓ 1 callers
Method
getAverageRegret
获取平均悔恨值 @return 平均悔恨值
src/main/java/io/leavesfly/tinydl/modality/rl/environment/MultiArmedBanditEnvironment.java:236
↓ 1 callers
Method
getAverageReward
获取平均奖励 @return 平均奖励
src/main/java/io/leavesfly/tinydl/modality/rl/environment/MultiArmedBanditEnvironment.java:227
↓ 1 callers
Method
getBestAccuracy
获取最佳训练准确率 @return 最佳训练准确率
src/main/java/io/leavesfly/tinydl/mlearning/Monitor.java:286
↓ 1 callers
Method
getBestEstimatedReward
获取当前最优臂的估计奖励 @return 最优臂估计奖励
src/main/java/io/leavesfly/tinydl/modality/rl/agent/BanditAgent.java:181
↓ 1 callers
Method
getBestLoss
获取最佳损失值 @return 最佳损失值
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:333
↓ 1 callers
Method
getBufferUsage
获取经验回放缓冲区使用率 @return 使用率
src/main/java/io/leavesfly/tinydl/modality/rl/agent/DQNAgent.java:339
↓ 1 callers
Method
getCurrentEpsilon
获取当前探索率 @return 探索率
src/main/java/io/leavesfly/tinydl/modality/rl/agent/DQNAgent.java:348
↓ 1 callers
Method
getDFF
获取前馈网络隐藏维度 @return 前馈网络隐藏维度
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:408
↓ 1 callers
Method
getDecoder
获取解码器 @return 解码器实例
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/Transformer.java:133
↓ 1 callers
Method
getDetailedInfo
获取详细信息 @return 详细信息字符串
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:534
↓ 1 callers
Method
getDotFunc
获取函数节点的DOT表示 @param node 函数节点 @return DOT格式的函数节点表示字符串
src/main/java/io/leavesfly/tinydl/utils/Uml.java:91
↓ 1 callers
Method
getDotNode
递归获取变量节点的DOT表示 @param variableNode 变量节点 @param dotGraph DOT图字符串构建器
src/main/java/io/leavesfly/tinydl/utils/Uml.java:41
↓ 1 callers
Method
getDotVar
获取变量节点的DOT表示 @param node 变量节点 @return DOT格式的节点表示字符串
src/main/java/io/leavesfly/tinydl/utils/Uml.java:70
↓ 1 callers
Method
getEpsilon
获取当前探索率 @return 探索率
src/main/java/io/leavesfly/tinydl/modality/rl/policy/EpsilonGreedyPolicy.java:171
↓ 1 callers
Method
getException
()
src/main/java/io/leavesfly/tinydl/mlearning/parallel/ParallelBatchProcessor.java:117
↓ 1 callers
Method
getExpertHiddenDim
()
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:508
↓ 1 callers
Method
getLoadBalancingLoss
获取负载均衡损失 这个损失可以添加到总的训练损失中,以鼓励专家的均匀使用 @return 负载均衡损失值
src/main/java/io/leavesfly/tinydl/modality/nlp/block/MoETransformerBlock.java:254
↓ 1 callers
Method
getLossValue
()
src/main/java/io/leavesfly/tinydl/mlearning/parallel/ParallelBatchProcessor.java:114
↓ 1 callers
Method
getMnistDir
()
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/MnistDataSet.java:162
↓ 1 callers
Method
getModelDetailedInfo
获取模型详细信息 @return 详细信息
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:316
↓ 1 callers
Method
getModelSummary
获取模型简要信息 @return 简要信息
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:305
↓ 1 callers
Method
getNumChannels
获取通道数
src/main/java/io/leavesfly/tinydl/nnet/layer/norm/BatchNorm.java:84
↓ 1 callers
Method
getNumHeads
获取注意力头数 @return 注意力头数
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:401
↓ 1 callers
Method
getOptimalActionRate
计算选择概率最高的臂被选择的频率 @return 最优选择频率
src/main/java/io/leavesfly/tinydl/modality/rl/agent/BanditAgent.java:190
↓ 1 callers
Method
getOptimalReward
获取最优奖励
src/main/java/io/leavesfly/tinydl/example/rl/MultiArmedBanditExample.java:268
↓ 1 callers
Method
getOrCreateOutput
获取或创建输出数组
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/Col2ImUtil.java:26
↓ 1 callers
Method
getOrCreateOutput
获取或创建输出数组
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/Im2ColUtil.java:27
↓ 1 callers
Method
getParallelThreadCount
获取并行线程数 @return 并行线程数
src/main/java/io/leavesfly/tinydl/mlearning/Trainer.java:387
↓ 1 callers
Method
getPositionEmbeddings
获取位置嵌入
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2TokenEmbedding.java:144
↓ 1 callers
Method
getSeqIndex
生成从0开始的连续整数对象序列 @param size 序列长度 @return 连续整数对象数组
src/main/java/io/leavesfly/tinydl/utils/Util.java:61
↓ 1 callers
Method
getSize
()
src/main/java/io/leavesfly/tinydl/mlearning/dataset/GPT2TextDataset.java:151
↓ 1 callers
Method
getStateDim
获取状态空间维度 @return 状态空间维度
src/main/java/io/leavesfly/tinydl/modality/rl/Environment.java:93
↓ 1 callers
Method
getTimeList
获取训练时间列表 @return 训练时间列表
src/main/java/io/leavesfly/tinydl/mlearning/Monitor.java:256
↓ 1 callers
Method
getToken
获取ID对应的token @param id 要查询的ID @return ID对应的token,如果不存在则返回UNK_TOKEN
src/main/java/io/leavesfly/tinydl/modality/nlp/SimpleTokenizer.java:328
↓ 1 callers
Method
getTokenEmbeddings
获取token嵌入
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2TokenEmbedding.java:115
↓ 1 callers
Method
getTrainImages
()
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/MnistDataSet.java:154
↓ 1 callers
Method
getTrainLabels
()
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/MnistDataSet.java:158
↓ 1 callers
Method
getTrainingTimeMs
获取训练时间(毫秒) @return 训练时间(毫秒)
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:445
↓ 1 callers
Method
getVocabSize
获取词汇表大小 @return 词汇表大小
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:380
↓ 1 callers
Method
getVocabSize
获取词汇表大小
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/Word2VecDataSet.java:218
↓ 1 callers
Method
getWeight
获取嵌入权重矩阵 @return 嵌入权重矩阵参数
src/main/java/io/leavesfly/tinydl/nnet/layer/embedd/Embedding.java:158
↓ 1 callers
Method
getYs
获取输出数据数组 @return 输出数据数组
src/main/java/io/leavesfly/tinydl/mlearning/dataset/ArrayDataset.java:130
↓ 1 callers
Method
getZScore
获取指定置信水平对应的Z分数(近似) @param confidenceLevel 置信水平 @return Z分数
src/main/java/io/leavesfly/tinydl/modality/rl/agent/ThompsonSamplingBanditAgent.java:331
↓ 1 callers
Method
goldStein
(Variable x, Variable y)
src/test/java/io/leavesfly/tinydl/test/TestFunc.java:37
↓ 1 callers
Method
hyperparameterSensitivityAnalysis
超参数敏感性分析
src/main/java/io/leavesfly/tinydl/example/rl/RLAlgorithmComparison.java:148
↓ 1 callers
Method
im2colFast
高效版本:使用一维数组存储的更快版本 适用于大批量数据处理
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/Im2ColUtil.java:108
↓ 1 callers
Method
init
初始化方法
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:180
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:227
↓ 1 callers
Method
init
初始化方法
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:167
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoELayer.java:136
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoEExpertNetwork.java:111
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoEGatingNetwork.java:78
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/modality/nlp/block/MoETransformerBlock.java:147
↓ 1 callers
Method
init
初始化层的参数
src/main/java/io/leavesfly/tinydl/nnet/LayerAble.java:53
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/norm/Flatten.java:40
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/DepthwiseSeparableConvLayer.java:69
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/PoolingLayer.java:83
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/cnn/ConvLayer.java:125
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/MultiHeadAttention.java:62
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/LayerNorm.java:49
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/PositionalEncoding.java:45
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/TransformerDecoderLayer.java:68
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/FeedForward.java:52
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/GPT2TokenEmbedding.java:60
↓ 1 callers
Method
init
()
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/TransformerEncoderLayer.java:59
← previous
next →
501–600 of 1,708, ranked by callers