mindspore.mint.bmm

mindspore.mint.bmm(input, mat2)[源代码]

对输入的两个tensor执行批量矩阵乘积。

\[\text{output} = \text{input} @ \text{mat2}\]

input 与 mat2 仅支持三维。

参数：

input (Tensor) - 输入的第一个tensor。
mat2 (Tensor) - 输入的第二个tensor。

返回：

Tensor

支持平台：

Ascend

样例：

>>> import mindspore
>>> a = mindspore.mint.ones([2, 3, 4])
>>> b = mindspore.mint.ones([2, 4, 5])
>>> mindspore.mint.bmm(a, b)
    Tensor(shape=[2, 3, 5], dtype=Float32, value=
    [[[ 4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00],
      [ 4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00],
      [ 4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00]],
     [[ 4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00],
      [ 4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00],
      [ 4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00,  4.00000000e+00]]])