执行动作，环境状态转移 @param action 智能体选择的动作 @return StepResult 包含下一状态、奖励、是否结束等信息

按指定形状进行压缩累加运算 <p>将当前数组按指定形状进行压缩，超出目标形状的部分会累加到对应位置</p> @param _shape 目标形状 @return 压缩累加结果数组 @throws IllegalArgumentException 当数组不是矩阵或形状不合法时抛出

将整型数组转换为浮点数组 @param src 整型数组 @return 浮点数组

获取架构类型 @return 架构类型

获取维度数量 @return 维度数量

获取最终损失值 @return 最终损失值

获取性能指标 @return 性能指标映射

获取总参数数量 @return 总参数数量

获取训练轮次 @return 训练轮次

计算损失值 @param y 真实标签 @param predict 预测值 @return 损失值变量

幂运算，对数组每个元素进行幂运算 @param number 幂指数 @return 幂运算结果数组

形状转字符串 @param shape 形状对象 @return 形状字符串表示

显示文件信息 @param filePath 文件路径 @param description 文件描述

减法运算 对当前变量与另一个变量执行减法运算 @param other 参与运算的另一个变量 @return 减法运算结果的新变量

求和运算 对变量的所有元素求和 @return 求和结果的新变量

切断计算图 用于RNN中切断计算图，防止梯度回传过长导致的梯度消失或爆炸问题。 该方法会清除当前变量的creator引用，并递归地对输入变量调用unChainBackward。

沿指定轴查找最大值的索引 @param axis 查找轴，axis=0表示按行查找每列的最大值索引，axis=1表示按列查找每行的最大值索引 @return 最大值索引数组 @throws IllegalArgumentException 当数组不是矩阵或轴参数无效时抛出

构建词汇表 @param corpus 语料库 (词的列表)

余弦函数运算，对数组每个元素进行cos运算 @return 余弦运算结果数组

获取动作空间维度 @return 动作空间维度

获取最佳训练损失值 @return 最佳训练损失值

获取输入形状 @return 输入形状

获取状态空间维度 @return 状态空间维度

优化版本：对四维输入数组执行 im2col 操作。 增加了缓存机制和性能优化 @param input 预期形状为 [numSamples][channels][height][width] 的四维数组。 @param filterH 滤波器高度。 @param filterW 滤波器宽度。

从经验中学习更新模型 @param experience 经验数据

绘制折线图 @param x X坐标数组 @param y Y坐标数组 @param name 数据系列名称

自然对数运算，对数组每个元素进行ln运算 @return 对数运算结果数组 @throws ArithmeticException 当输入值小于等于0时抛出

取反操作，对数组每个元素取负值 @return 取反后的数组

取反操作 对变量执行取反运算，返回一个新的变量，其值为原变量值的相反数。 @return 取反后的变量

渲染环境（可选实现）

将数据随机打散

正弦函数运算，对数组每个元素进行sin运算 @return 正弦运算结果数组

Softmax函数运算，按行计算概率分布 <p>Softmax函数公式：softmax(x_i) = exp(x_i) / Σ(exp(x_j))</p> <p>使用数值稳定版本实现，避免指数运算溢出</p> @return Softmax运算结果数组 @throws IllegalArgume

平方运算，对数组每个元素进行平方运算 @return 平方运算结果数组

将浮点数组转换为整型数组 @param src 浮点数组 @return 整型数组

在指定位置累加数组元素 <p>在指定的行和列位置上累加另一个数组的元素。这个方法常用于反向传播中梯度的累积。</p> <p>使用示例：</p> <pre> NdArray a = new NdArray(new float[][]{{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}

添加性能指标 @param metricName 指标名称 @param value 指标值

添加token到词汇表 @param token 要添加的token @param id token对应的ID

除法运算 对当前变量与另一个变量执行除法运算 @param other 参与运算的另一个变量 @return 除法运算结果的新变量

将文本编码为token ID数组 @param text 输入文本 @param addSpecialTokens 是否添加特殊token @return token ID数组

自然指数运算，对数组每个元素进行e为底的指数运算 @return 指数运算结果数组

设置每个块的滤波器数量 @param nums 滤波器数量数组 @return 当前配置对象

索引操作 根据指定的行列索引获取变量的子集 @param _rowSlices 行索引数组 @param _colSlices 列索引数组 @return 索引操作结果的新变量

获取总层数 @return 总层数

获取输入数据数组 @return 输入数据数组

矩阵全元素大于比较，判断当前数组是否所有元素都大于另一个数组对应元素 @param other 另一个操作数数组 @return 比较结果，true表示所有元素都大于，false表示存在不大于的元素 @throws IllegalArgumentException 当两个数组形状不一致时抛出

对数运算 对变量执行自然对数运算(ln(x)) @return 对数运算结果的新变量

获取下一对数据 @return 数据对，如果已遍历完则返回null

重置模型状态 主要用于RNN等有状态的模型，清除历史状态信息

绘制散点图 @param x X坐标数组 @param y Y坐标数组

显示图表

Sigmoid函数运算，对数组每个元素进行sigmoid运算 <p>Sigmoid函数公式：f(x) = 1 / (1 + e^(-x))</p> @return Sigmoid运算结果数组

双曲正切运算 对变量执行双曲正切运算 @return 双曲正切运算结果的新变量

通用的一元运算方法，对数组每个元素进行一元运算 @param operation 一元运算操作函数 @return 运算结果数组

通用的二元运算方法，对两个相同形状的数组进行元素级运算 @param other 另一个操作数数组 @param operation 二元运算操作函数 @param operationName 操作名称，用于错误提示 @return 运算结果数组 @throws IllegalArgumentEx

创建用于MNIST数据集的深度卷积网络 输入：28x28x1，输出：10个类别 @return 构建好的卷积网络

计算指定臂的UCB值 UCB(i) = Q(i) + c sqrt(ln(t) / N(i)) @param armIndex 臂索引 @return UCB值

优化版本：将列格式的数据还原成多维图像数组。 增加了缓存机制和性能优化 @param col 列格式的数据 @param imgShape 原始图像数据的形状，形式为 [N, C, H, W]。 @param filterH 滤波器的高度。 @param filterW 滤波器的宽度

收集训练信息（损失值） @param loss 当前批次的损失值

创建随机序列数据（用于测试） @param batchSize 批次大小 @param seqLen 序列长度 @param vocabSize 词汇表大小 @return 随机序列数据

创建微型GPT-2模型的构造函数 用于快速实验和调试 @param name 模型名称 @param vocabSize 词汇表大小 @return GPT-2模型实例

数组除法运算，对应元素相除 @param other 另一个操作数数组 @return 除法运算结果 @throws IllegalArgumentException 当两个数组形状不一致时抛出 @throws ArithmeticException 当除数接近0时抛出

设置Dropout率 @param rate Dropout率 @return 当前配置对象

相等比较运算，比较两个数组对应元素是否相等 @param other 另一个操作数数组 @return 比较结果数组，1.0表示相等，0.0表示不相等 @throws IllegalArgumentException 当两个数组形状不一致时抛出

指数运算 对变量执行自然指数运算(e^x) @return 指数运算结果的新变量

创建指定形状的单位矩阵（对角矩阵） @param shape 矩阵形状（必须为方形矩阵） @return 单位矩阵 @throws IllegalArgumentException 当形状不是矩阵或不是方形矩阵时抛出

设置全连接层隐藏单元数 @param size 隐藏单元数 @return 当前配置对象

格式化日期 @param date 日期对象 @return 格式化后的日期字符串

生成训练样本 @param corpus 语料库 @return 训练样本列表

获取当前最优臂的索引（基于估计奖励） @return 最优臂索引

获取当前状态 @return 当前状态

获取模型维度 @return 模型维度

获取模型描述 @return 模型描述

获取指定维度的大小 @param dimIndex 维度索引 @return 指定维度的大小 @throws IndexOutOfBoundsException 当维度索引超出范围时抛出异常

获取变量节点的DOT图表示 @param variableNode 变量节点 @return DOT格式的图表示字符串

获取环境信息（用于调试和监控） @return 环境信息字典

获取模型名称 @return 模型名称

获取Transformer层数 @return Transformer层数

获取词向量 @param word 词 @return 词向量

初始化训练器 @param _dataSet 数据集 @param _model 模型 @param _loss 损失函数 @param _optimizer 优化器

检查位置是否是障碍物 @param x X坐标 @param y Y坐标 @return 是否是障碍物

获取变量是否需要计算梯度 @return 是否需要计算梯度

检查位置是否有效 @param x X坐标 @param y Y坐标 @return 是否有效

线性层的前向传播方法 @param inputs 输入变量数组，通常只包含一个输入变量 @return 线性变换后的输出变量

使用ModelSerializer加载模型（推荐使用） @param filePath 模型文件路径 @return 加载的模型

矩阵乘法 对当前变量与另一个变量执行矩阵乘法运算 @param other 参与运算的另一个变量 @return 矩阵乘法结果的新变量

矩阵均值运算，沿指定轴计算均值 @param axis 聚合轴，axis=0表示按列计算均值，axis=1表示按行计算均值 @return 均值运算结果数组

将token ID列表填充到指定长度 @param tokenIds 原始token ID列表 @param maxLength 目标长度 @param padding 填充方式（"pre"或"post"） @return 填充后的数组

打印训练信息

将二维矩阵重塑回三维张量

将三维张量重塑为二维矩阵以用于线性变换

正弦运算 对变量执行正弦运算 @return 正弦运算结果的新变量

求和到指定形状 将变量求和到指定形状 @param shape 目标形状 @return 求和后的新变量

训练并评估智能体 @param agent 智能体（可以是DQN或REINFORCE） @param env 环境 @param numEpisodes 训练回合数 @param isReinforce 是否为REINFORCE算法 @return 平均评估奖励