mindspore.ops.Einsum ===================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/ops/mindspore.ops.Einsum.rst :alt: 查看源文件 .. py:class:: mindspore.ops.Einsum(equation) 此算子使用爱因斯坦求和约定(Einsum)进行Tensor计算。支持对角线、约和、转置、矩阵乘、乘积、内积运算等。 参数: - **equation** (str) - 属性,表示要执行的计算。该值只能使用letter([a-z][A-Z])、commas(,)、ellipsis(...)和arrow(->)。letter([a-z][A-Z])表示输入的Tensor的维度,commas(,)表示Tensor维度之间的分隔符,ellipsis(...)表示不关心的Tensor维度,arrow(->)的左侧表示输入Tensor,右侧表示所需的输出维度。 输入: - **x** (Union[tuple(Tensor), list(Tensor)]) - 用于计算的输入Tensor,输入必须是Tensor的tuple/list。当输入只有一个Tensor时,可以输入(Tensor, )。支持数据类型float16、float32、float64,输入各Tensor的数据类型必须相同。 输出: Tensor,shape可以从方程中获得,数据类型与输入Tensor相同。 异常: - **TypeError** - 如果 `equation` 本身无效,或者 `equation` 与输入Tensor不匹配。 - **TypeError** - 如果输入Tensor的数据类型不一致或者数据类型不是float16、float32、float64。