2022-12-01 16:57:52 +08:00
|
|
|
|
mindspore.ops.einsum
|
|
|
|
|
|
====================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:function:: mindspore.ops.einsum(equation, *operands)
|
|
|
|
|
|
|
|
|
|
|
|
基于爱因斯坦求和约定（Einsum）符号，指定维度对输入Tensor元素的乘积求和。你可以使用这个运算符来执行对角线、减法、转置、矩阵乘法、乘法、内积运算等等。
|
|
|
|
|
|
|
|
|
|
|
|
参数：
|
2022-12-14 16:23:12 +08:00
|
|
|
|
- **equation** (str) - 基于爱因斯坦求和约定的符号，表示想要执行的操作。符号只能包含字母、逗号、省略号和箭头。字母表示输入Tensor维数，逗号表示单独的Tensor，省略号表示忽略的Tensor维数，箭头的左边表示输入Tensor，右边表示期望输出的维度。
|
|
|
|
|
|
- **operands** (Tensor) - 用于计算的输入Tensor。Tensor的数据类型必须相同。
|
2022-12-01 16:57:52 +08:00
|
|
|
|
|
|
|
|
|
|
返回：
|
|
|
|
|
|
Tensor，shape可以根据 `equation` 得到。数据类型和输入Tensor相同。
|
|
|
|
|
|
|
|
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - `equation` 无效或者不匹配输入Tensor。
|
