mindspore.mint.bmm
- mindspore.mint.bmm(input, mat2)[source]
Performs batch matrix-matrix multiplication of two three-dimensional tensors.
\[\text{output}= \text{input} @ \text{mat2}\]- Parameters
- Returns
Tensor, the output tensor of shape (b, n, p), where each matrix is the product of the corresponding matrices in the input batches.
- Raises
ValueError – If input or mat2 is not three-dimensional tensors.
ValueError – If the length of the third dimension of input is not equal to the length of the second dimension of mat2.
ValueError – If the batch size of the inputs is not equal to the batch size of the mat2.
- Supported Platforms:
Ascend
Examples
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor >>> from mindspore import mint >>> a = Tensor(np.ones(shape=[2, 3, 4]), mindspore.float32) >>> b = Tensor(np.ones(shape=[2, 4, 5]), mindspore.float32) >>> output = mint.bmm(a, b) >>> print(output) [[[4. 4. 4. 4. 4.] [4. 4. 4. 4. 4.] [4. 4. 4. 4. 4.]] [[4. 4. 4. 4. 4.] [4. 4. 4. 4. 4.] [4. 4. 4. 4. 4.]]]