mindspore.ops.Einsum
=====================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/r2.3.1/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/r2.3.1/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。