Fonction torch.mm de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.mmest une fonction de PyTorch utilisée pour effectuer une multiplication matricielle bidimensionnelle.

Définition de fonction

torch.mm(input, mat2, out)

Exemple d'utilisation

Exemple

import torch

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

# Multiplication matricielle
C = torch.mm(A, B)

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

Le résultat de sortie est:

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

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions