Función PyTorch torch.linalg.matmul

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

torch.linalg.matmulEs una función del módulo de álgebra lineal de PyTorch utilizada para realizar la multiplicación de matrices. Estorch.matmulun alias de, que proporciona una interfaz de álgebra lineal más unificada.

Definición de la función

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

Parámetros:

  • input(Tensor): El primer tensor de entrada.
  • other(Tensor): El segundo tensor de entrada.
  • out(Tensor, opcional): El tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el resultado de la multiplicación de matrices.

Ejemplo de uso

Ejemplo

import torch

# Crear dos matrices
A = torch.randn(3, 4)
B = torch.randn(4, 5)

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

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

El resultado de salida es:

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

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

Otras extensiones