Fonction torch.linalg.matmul de PyTorch

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

torch.linalg.matmulest une fonction du module d'algèbre linéaire de PyTorch utilisée pour effectuer la multiplication matricielle. C'esttorch.matmulun alias qui fournit une interface d'algèbre linéaire plus unifiée.

Définition de fonction

torch.linalg.matmul(input, other, out=None)

Parameters:

  • input(Tensor) : Le premier tenseur d'entrée.
  • other(Tensor) : Le deuxième tenseur d'entrée.
  • out(Tensor, optionnel) : Le tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie le résultat de la multiplication matricielle.

Exemple d'utilisation

Exemple

import torch

# Créer deux matrices
A = torch.randn(3, 4)
B = torch.randn(4, 5)

# Multiplication matricielle
C = torch.linalg.matmul(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([3, 4])
B 的形状: torch.Size([4, 5])
C 的形状: torch.Size([3, 5])

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

Autres extensions