复数爱因斯坦求和
复数操作
复数爱因斯坦求和
复数爱因斯坦求和(Complex Einstein Summation)算子用于对一组输入张量进行爱因斯坦求和,输入张量允许为复数。
参数列表
| 参数 | 类型 | 参数描述 |
|---|---|---|
| 求和标记 | 输入 | 字符串类型,是这样的形式:"i,j->ij" |
| 输入张量 | 输入 | 进行求和的张量,可能有1个或多个操作数 |
| 输出张量 | 输出 | 表示计算结果,与输入张量精度一致 |
函数返回值
- 没有错误:操作成功。
- 对象未初始化:表示输入张量没有初始化。
- 类型不匹配:表示参数的数据类型不一致。
- 分配失败:表张量分配空间不足。
- 内部错误:表示内部调用出错。
参考来源:团体标准——《人工智能 科学计算 共性算子功能要求》——章节7.3.2.7
硬件验证状态
昇腾 NPU 1.8ms鲲鹏 CPU 2.6ms海光 CPU 2.1ms昆仑 XPU 2.4ms海光 DCU 2.3ms飞腾 CPU 3.1ms燧原 XPU 2.7ms沐曦 GPU 2.5ms
