mindspore.ops.MatMul
- class mindspore.ops.MatMul(transpose_a=False, transpose_b=False)[source]
Multiplies matrix x and matrix y.
where the
indicates the output of the i-th row and j-th column element.- Parameters
- Inputs:
x (Tensor) - The first tensor to be multiplied. The shape of the tensor is
. If transpose_x is True, its shape must be after transpose.y (Tensor) - The second tensor to be multiplied. The shape of the tensor is
. If transpose_y is True, its shape must be after transpose.
- Outputs:
Tensor, the shape of the output tensor is
.
- Raises
TypeError – If transpose_a or transpose_b is not a bool.
ValueError – If the column of matrix dimensions of x is not equal to the row of matrix dimensions of y.
ValueError – If length of shape of x or y is not equal to 2.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> x = Tensor(np.ones(shape=[1, 3]), mindspore.float32) >>> y = Tensor(np.ones(shape=[3, 4]), mindspore.float32) >>> matmul = ops.MatMul() >>> output = matmul(x, y) >>> print(output) [[3. 3. 3. 3.]]