PyTorch .mm()

In PyTorch, the .mm() method calculates the matrix product of two given tensors.

Syntax

torch.mm(ten1, ten2, *, out=None)

• ten1: The first tensor to be multiplied.
• ten2: The second tensor to be multiplied.
• out (Optional): The output tensor to be used. The default value is None.

Note: If ten1 is a (m x n) tensor and ten2 is a (n x p) tensor, then out will be a (m x p) tensor.

Example

The following example demonstrates the usage of the .mm() method:

import torch

# Define two tensors
ten1 = torch.tensor([[1, 2, 3],
                     [4, 3, 8],
                     [1, 7, 2]])

ten2 = torch.tensor([[2, 4, 1],
                     [1, 3, 6],
                     [2, 6, 5]])

# Multiply the tensors
out = torch.mm(ten1, ten2)

print(out)

Copy to clipboard
Copy to clipboard

The above code produces the following output:

tensor([[10, 28, 28],
        [27, 73, 62],
        [13, 37, 53]])

Copy to clipboard
Copy to clipboard

Codebyte Example

The following codebyte example shows the use of the .mm() method:

Visit us
Visit us
Hide code
Code
Output
Hide output
Copy to your clipboard
Copy Codebyte
Run