Función torch.mm de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.mmEs la función en PyTorch utilizada para realizar la multiplicación de matrices bidimensionales.

Definición de la función

torch.mm(input, mat2, out)

Ejemplo de uso

Ejemplo

import torch

A = torch.randn(2, 3)
B = torch.randn(3, 4)

# Multiplicación de matrices
C = torch.mm(A, B)

print("Forma de A:", A.shape)
print("Forma de B:", B.shape)
print("Forma de C:", C.shape)

El resultado de la salida es:

A 形状: torch.Size([2, 3])
B 形状: torch.Size([3, 4])
C 形状: torch.Size([2, 4])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones