张量均值
高阶微分操作
张量均值
计算张量中所有元素的算术平均值，或沿指定维度计算均值。作为高阶微分库的基础算子，支持自动微分，常用于构造损失函数（如 MSE）或归一化层。
参数
| 参数名 | 类型 | 说明 |
|---|---|---|
input | Tensor | 输入张量 |
dim | int 或 list[int]，可选 | 求均值的维度；缺省时对全部元素求均值 |
keepdim | bool | 是否保持输出维度 |
返回值
| 类型 | 说明 |
|---|---|
| Tensor | 均值标量或张量 |
说明
- 在自动微分环境下，均值算子的梯度会平均分配给所有参与计算的元素。
- 在高阶微分（如雅可比矩阵、海森矩阵）中，均值为线性变换，其二阶导数为零，常用于简化模型输出后再求导。
标准来源
国家标准 20252061-Z-469 高阶微分操作章节
硬件验证状态
昇腾 NPU 1.8ms鲲鹏 CPU 2.6ms海光 CPU 2.1ms昆仑 XPU 2.4ms海光 DCU 2.3ms飞腾 CPU 3.1ms燧原 XPU 2.7ms沐曦 GPU 2.5ms