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
↓ 434 callers
Method
getValue
()
src/main/java/io/leavesfly/tinydl/func/Variable.java:233
↓ 262 callers
Method
getShape
获取数组的形状 @return 数组形状
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1578
↓ 210 callers
Method
getMatrix
将数组转换为二维数组(矩阵)返回 @return 二维数组表示 @throws IllegalArgumentException 当数组维度大于2时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1603
↓ 189 callers
Method
get
按维度下标获取某一个值 @param _dimension 维度下标数组 @return 对应位置的值 @throws IllegalArgumentException 当维度数量不匹配时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1792
↓ 168 callers
Method
add
数组加法运算,对应元素相加 @param other 另一个操作数数组 @return 加法运算结果 @throws IllegalArgumentException 当两个数组形状不一致时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:446
↓ 139 callers
Method
getGrad
()
src/main/java/io/leavesfly/tinydl/func/Variable.java:245
↓ 124 callers
Method
format
格式化浮点数,保留适当的小数位数 @param num 要格式化的数字 @return 格式化后的字符串
src/main/java/io/leavesfly/tinydl/utils/Util.java:131
↓ 120 callers
Method
call
函数的执行函数,执行函数的前向传播计算并构建计算图 该方法执行以下操作: 1. 验证输入变量数量是否符合要求 2. 从输入变量中提取NdArray值 3. 调用forward方法执行前向传播计算 4. 创建输出变量 5. 在训练模式下构建计算图 @param _inputs 输入变量数组 @re
src/main/java/io/leavesfly/tinydl/func/Function.java:45
↓ 110 callers
Method
getColumn
获取列数(仅适用于二维形状) @return 列数 @throws IllegalStateException 当形状不是二维时抛出异常
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:93
↓ 87 callers
Method
toString
()
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:209
↓ 84 callers
Method
getNumber
获取数组的第一个元素值(标量值) @return 第一个元素值
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1569
↓ 80 callers
Method
getName
获取层的名称 @return 层的名称
src/main/java/io/leavesfly/tinydl/nnet/LayerAble.java:76
↓ 76 callers
Method
setName
(String name)
src/main/java/io/leavesfly/tinydl/func/Variable.java:275
↓ 75 callers
Method
size
获取缓冲区当前大小 @return 当前存储的经验数量
src/main/java/io/leavesfly/tinydl/modality/rl/ReplayBuffer.java:104
↓ 69 callers
Method
size
计算对应形状的N维数组的元素总数 @return 元素总数
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:132
↓ 66 callers
Method
backward
变量的反向传播(递归实现) 根据正向传播时构建的计算图,从当前变量开始反向传播计算每个变量的梯度。 如果变量不需要计算梯度,则直接返回。 如果梯度未初始化,则初始化为全1的数组。 然后递归地调用生成该变量的函数的backward方法计算输入变量的梯度。
src/main/java/io/leavesfly/tinydl/func/Variable.java:120
↓ 65 callers
Method
addLayer
向Block中添加一个Layer @param layerAble 要添加的Layer实例
src/main/java/io/leavesfly/tinydl/nnet/Block.java:71
↓ 60 callers
Method
getRow
获取行数(仅适用于二维形状) @return 行数 @throws IllegalStateException 当形状不是二维时抛出异常
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:80
↓ 60 callers
Method
writeJsonField
写入JSON字段 @param writer PrintWriter对象 @param key 字段键 @param value 字段值 @param hasNext 是否有后续字段
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfoExporter.java:297
↓ 58 callers
Method
min
沿指定轴查找最小值 @param axis 查找轴,axis=0表示按行查找每列的最小值,axis=1表示按列查找每行的最小值 @return 最小值数组 @throws IllegalArgumentException 当数组不是矩阵或轴参数无效时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1300
↓ 56 callers
Method
equals
(Object o)
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:190
↓ 53 callers
Method
dot
矩阵内积运算(矩阵乘法) <p>执行标准的矩阵乘法运算,要求第一个矩阵的列数等于第二个矩阵的行数</p> @param other 另一个矩阵 @return 矩阵乘法结果 @throws IllegalArgumentException 当数组不是矩阵或维度不匹配时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1148
↓ 52 callers
Method
toString
优化的toString方法,提供数组的字符串表示 <p>对于小数组会显示所有元素,对于大数组只会显示部分元素</p> @return 数组的字符串表示
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1680
↓ 50 callers
Method
max
沿指定轴查找最大值 @param axis 查找轴,axis=0表示按行查找每列的最大值,axis=1表示按列查找每行的最大值 @return 最大值数组 @throws IllegalArgumentException 当数组不是矩阵或轴参数无效时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1258
↓ 49 callers
Method
zeros
创建指定形状的全零数组 @param shape 数组形状 @return 全零数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:256
↓ 47 callers
Method
getIndex
根据多维索引计算一维数组中的位置 @param indices 多维索引 @return 一维数组中的位置 @throws IllegalArgumentException 当索引维度与形状维度不匹配时抛出异常 @throws IndexOutOfBoundsException 当索引超出范围时抛
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:148
↓ 46 callers
Method
set
按维度下标设置某一个值 @param value 要设置的值 @param _dimension 维度下标数组 @throws IllegalArgumentException 当维度数量不匹配时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1776
↓ 42 callers
Method
transpose
矩阵转置操作(二维矩阵),行列互换 @return 转置后的矩阵 @throws IllegalArgumentException 当数组不是矩阵时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:800
↓ 41 callers
Method
addParam
添加参数到层中 @param paramName 参数名称 @param value 参数值
src/main/java/io/leavesfly/tinydl/nnet/LayerAble.java:95
↓ 38 callers
Method
isMatrix
判断是否是矩阵(二维形状) @return 如果是二维形状返回true,否则返回false
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:105
↓ 37 callers
Method
likeRandomN
创建标准正态分布(均值为0,标准差为1)的随机数组 @param shape 数组形状 @return 标准正态分布随机数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:318
↓ 36 callers
Method
sqrt
平方根运算,对数组每个元素进行开方运算 @return 平方根运算结果数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:662
↓ 35 callers
Method
mulNum
数组与标量相乘 @param number 标量值 @return 乘法运算结果
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:478
↓ 35 callers
Method
reshape
数组变形操作,改变数组形状但保持元素总数不变 @param newShape 新的数组形状 @return 变形后的数组 @throws IllegalArgumentException 当新形状大小与原形状不匹配时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:892
↓ 34 callers
Method
ones
创建指定形状的全一数组 @param shape 数组形状 @return 全一数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:266
↓ 33 callers
Method
mul
数组乘法运算,对应元素相乘 @param other 另一个操作数数组 @return 乘法运算结果 @throws IllegalArgumentException 当两个数组形状不一致时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:468
↓ 32 callers
Method
isEmpty
检查缓冲区是否为空 @return 是否为空
src/main/java/io/leavesfly/tinydl/modality/rl/ReplayBuffer.java:122
↓ 32 callers
Method
sub
数组减法运算,对应元素相减 @param other 另一个操作数数组 @return 减法运算结果 @throws IllegalArgumentException 当两个数组形状不一致时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:457
↓ 30 callers
Method
mul
乘法运算 对当前变量与另一个变量执行乘法运算 @param other 参与运算的另一个变量 @return 乘法运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:318
↓ 27 callers
Method
linear
线性变换 对变量执行线性变换 y = xW + b @param w 权重变量 @param b 偏置变量,可为null @return 线性变换结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:604
↓ 26 callers
Method
requireInputNum
()
src/test/java/io/leavesfly/tinydl/test/func/FunctionOptimizationTest.java:47
↓ 24 callers
Method
getOutputShape
获取输出形状 @return 输出形状
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:221
↓ 24 callers
Method
updateLastModifiedTime
更新最后修改时间
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:507
↓ 23 callers
Method
getParamBy
根据参数名称获取参数 @param paramName 参数名称 @return 对应的参数对象
src/main/java/io/leavesfly/tinydl/nnet/LayerAble.java:105
↓ 23 callers
Method
merge
按照指定轴对多个NdArray进行合并 @param axis 合并的轴向,0表示按第一个维度合并,1表示按第二个维度合并,以此类推 @param ndArrays 需要合并的NdArray数组 @return 合并后的NdArray @throws IllegalArgumentExceptio
src/main/java/io/leavesfly/tinydl/ndarr/NdArrayUtil.java:18
↓ 22 callers
Method
abs
绝对值运算,对数组每个元素取绝对值 @return 绝对值数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:565
↓ 22 callers
Method
add
加法运算 对当前变量与另一个变量执行加法运算 @param other 参与运算的另一个变量 @return 加法运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:292
↓ 22 callers
Method
setRequireGrad
(boolean _requireGrad)
src/main/java/io/leavesfly/tinydl/func/Variable.java:98
↓ 20 callers
Method
getCreator
()
src/main/java/io/leavesfly/tinydl/func/Variable.java:263
↓ 17 callers
Method
hashCode
()
src/main/java/io/leavesfly/tinydl/ndarr/Shape.java:198
↓ 16 callers
Method
like
创建指定形状和值的数组 @param shape 数组形状 @param value 填充值 @return 指定值填充的数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:296
↓ 16 callers
Method
toVariableX
将输入数据转换为Variable对象 为了提高性能,该方法会缓存Variable实例 @return 输入数据的Variable表示
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:85
↓ 15 callers
Method
forward
模型前向传播 @param inputs 输入变量 @return 输出变量
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:197
↓ 15 callers
Method
getInputs
获取函数的输入变量数组 @return 输入变量数组
src/main/java/io/leavesfly/tinydl/func/Function.java:101
↓ 15 callers
Method
getReward
获取奖励 @return 奖励值
src/main/java/io/leavesfly/tinydl/modality/rl/Experience.java:101
↓ 15 callers
Method
getSize
获取批次大小 @return 批次大小
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:76
↓ 14 callers
Method
broadcastTo
数组广播运算,将当前数组广播到指定形状 <p>广播机制允许小数组与大数组进行运算,小数组会重复填充以匹配大数组的形状</p> @param _shape 目标广播形状 @return 广播结果数组 @throws IllegalArgumentException 当数组不是矩阵或形状不合法时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1072
↓ 14 callers
Method
getAllParams
获取所有参数 @return 参数映射
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:213
↓ 14 callers
Method
getOutputShape
获取输出数据的形状 @return 输出形状
src/main/java/io/leavesfly/tinydl/nnet/LayerAble.java:123
↓ 14 callers
Method
sum
元素累和运算,计算数组所有元素的总和 @return 所有元素的总和(标量)
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:922
↓ 14 callers
Method
toString
()
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:599
↓ 14 callers
Method
toVariableY
将标签数据转换为Variable对象 为了提高性能,该方法会缓存Variable实例 @return 标签数据的Variable表示
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:97
↓ 14 callers
Method
update
更新所有参数
src/main/java/io/leavesfly/tinydl/mlearning/optimize/Adam.java:65
↓ 13 callers
Method
broadcastTo
广播操作 将变量广播到指定形状 @param shape 目标形状 @return 广播后的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:527
↓ 13 callers
Method
getBatches
获取批次数据列表 @return 批次数据列表
src/main/java/io/leavesfly/tinydl/mlearning/dataset/DataSet.java:39
↓ 13 callers
Method
isDone
判断是否为终止状态 @return 是否为终止状态
src/main/java/io/leavesfly/tinydl/modality/rl/Experience.java:119
↓ 12 callers
Method
clone
克隆专家网络(用于创建具有相同结构但不同参数的专家) @param newName 新专家的名称 @param newExpertId 新专家的ID @return 新的专家网络实例
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoEExpertNetwork.java:224
↓ 12 callers
Method
prepare
准备数据集 该方法确保数据集只被准备一次
src/main/java/io/leavesfly/tinydl/mlearning/dataset/DataSet.java:46
↓ 12 callers
Method
selectAction
根据当前状态选择动作 @param state 当前状态 @return 选择的动作
src/main/java/io/leavesfly/tinydl/modality/rl/Agent.java:94
↓ 12 callers
Method
setGrad
(NdArray _grad)
src/main/java/io/leavesfly/tinydl/func/Variable.java:249
↓ 11 callers
Method
clearGrads
清除梯度 在每次反向传播前调用,清除历史梯度信息
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:205
↓ 11 callers
Method
getModelInfo
获取模型信息 @return 模型信息
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:248
↓ 11 callers
Method
getName
获取模型名称 @return 模型名称
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:230
↓ 11 callers
Method
getNextState
获取下一状态 @return 下一状态
src/main/java/io/leavesfly/tinydl/modality/rl/Experience.java:110
↓ 11 callers
Method
getOutput
获取函数的输出变量 @return 输出变量
src/main/java/io/leavesfly/tinydl/func/Function.java:119
↓ 11 callers
Method
getSeq
生成从0开始的连续整数序列 @param size 序列长度 @return 连续整数数组
src/main/java/io/leavesfly/tinydl/utils/Util.java:47
↓ 11 callers
Method
getWord2idx
获取词到索引的映射 @return 词到索引的映射
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:484
↓ 11 callers
Method
layerForward
前向传播 @param inputs 输入变量数组 @return 前向传播结果
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:264
↓ 11 callers
Method
reset
重置智能体状态
src/main/java/io/leavesfly/tinydl/modality/rl/Agent.java:247
↓ 11 callers
Method
tanh
双曲正切函数运算,对数组每个元素进行tanh运算 @return 双曲正切运算结果数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:698
↓ 11 callers
Method
transpose
转置操作 对变量执行转置操作 @return 转置后的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:590
↓ 10 callers
Method
clear
清空缓冲区
src/main/java/io/leavesfly/tinydl/modality/rl/ReplayBuffer.java:138
↓ 10 callers
Method
getItem
获取数组的子集(切片操作) @param _rowSlices 行索引数组,null表示选择所有行 @param _colSlices 列索引数组,null表示选择所有列 @return 切片结果数组 @throws IllegalArgumentException 当数组不是矩阵或参数不合法时抛
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1185
↓ 10 callers
Method
getMaxSeqLength
获取最大序列长度 @return 最大序列长度
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:415
↓ 10 callers
Method
likeRandom
创建指定范围内的均匀分布随机数组 @param min 最小值(包含) @param max 最大值(包含) @param shape 数组形状 @return 均匀分布随机数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:346
↓ 10 callers
Method
setTraining
设置训练模式 @param training 是否为训练模式
src/main/java/io/leavesfly/tinydl/modality/rl/Agent.java:143
↓ 10 callers
Method
setValue
(NdArray value)
src/main/java/io/leavesfly/tinydl/func/Variable.java:241
↓ 9 callers
Method
benchmarkForward
基准测试辅助方法
src/test/java/io/leavesfly/tinydl/test/cnn/CnnPerformanceBenchmark.java:251
↓ 9 callers
Method
compareParameters
比较两个模型的参数 @param model1 模型1 @param model2 模型2 @param tolerance 容忍度 @return 参数是否相同
src/main/java/io/leavesfly/tinydl/mlearning/ParameterManager.java:135
↓ 9 callers
Method
getParams
获取层的所有参数 @return 参数映射表
src/main/java/io/leavesfly/tinydl/nnet/LayerAble.java:85
↓ 9 callers
Method
hasNext
检查是否还有更多数据 @return 如果还有数据返回true,否则返回false
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:121
↓ 9 callers
Method
squ
平方运算 对变量执行平方运算 @return 平方运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:359
↓ 8 callers
Method
clearGrad
清理梯度 将变量的梯度设置为null,释放梯度占用的内存。 通常在每次训练迭代开始前调用,以确保梯度不会累积。
src/main/java/io/leavesfly/tinydl/func/Variable.java:229
↓ 8 callers
Method
createTestModel
创建测试模型
src/test/java/io/leavesfly/tinydl/test/ModelSerializationTest.java:281
↓ 8 callers
Method
divNum
数组与标量相除 @param number 标量值 @return 除法运算结果 @throws ArithmeticException 当除数为0时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:506
↓ 8 callers
Method
forward
(NdArray... inputs)
src/test/java/io/leavesfly/tinydl/test/func/FunctionOptimizationTest.java:37
↓ 8 callers
Method
getCurrentIndex
获取当前遍历索引 @return 当前索引
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:136
↓ 8 callers
Method
getTrainDataSet
获取训练数据集 @return 训练数据集
src/main/java/io/leavesfly/tinydl/mlearning/dataset/DataSet.java:77
↓ 8 callers
Method
mathOperation
通用的数学函数运算方法,对数组每个元素应用数学函数 @param mathFunc 数学函数操作 @return 运算结果数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:629
↓ 8 callers
Method
printMatrix
(NdArray array)
src/test/java/io/leavesfly/tinydl/test/ndarr/NdArrayTodoVerification.java:97
next →
1–100 of 1,708, ranked by callers