逐元素乘法

高阶微分操作

逐元素乘法

逐元素乘法(Element-wise Multiplication)算子正向传播对两个张量逐元素求乘积 C = A * B。

反向传播对逐元素乘法操作的结果 C,计算其对 A 和 B 的导数。

正向传播参数列表

参数类型参数描述
输入张量输入元素类型可以为整数、浮点数
输出张量输出与第一个输入张量在计算上兼容
输出张量输出表示计算结果

反向传播参数列表

参数类型参数描述
输入张量输入关于输出张量C的梯度dC:从网络后续部分传回的梯度
输出张量输出关于输入张量A的梯度,计算为 dC * B
输出张量输出关于输入张量B的梯度,计算为 dC * A

函数返回值

  • 没有错误:操作成功。
  • 对象未初始化:表示输入张量没有初始化。
  • 类型不匹配:表示参数的数据类型不一致。
  • 分配失败:表张量分配空间不足。
  • 内部错误:表示内部调用出错。

注:正向传播参考IEEE 2941.1-2022标准中的张量乘法操作接口。

参考来源:团体标准——《人工智能 科学计算 共性算子功能要求》——章节7.3.1.5

硬件验证状态

昇腾 NPU 1.8ms鲲鹏 CPU 2.6ms海光 CPU 2.1ms昆仑 XPU 2.4ms海光 DCU 2.3ms飞腾 CPU 3.1ms燧原 XPU 2.7ms沐曦 GPU 2.5ms