bmm

    对输入x及输入y进行矩阵相乘。

    两个输入的维度必须等于3,并且矩阵x和矩阵y的第一维必须相等

    同时矩阵x的第二维必须等于矩阵y的第三维

    参数

    返回

    返回类型

    • Tensor。

    示例: