Функция torch.linalg.matmul в PyTorch
Справочное руководство Pytorch torch
torch.linalg.matmulявляется функцией в модуле линейной алгебры PyTorch, используемой для выполнения умножения матриц. Она являетсяtorch.matmulпсевдонимом, который обеспечивает более унифицированный интерфейс линейной алгебры.
Определение функции
torch.linalg.matmul(input, other, out=None)
Parameters:
input(Tensor): первый входной тензор.other(Tensor): Второй входной тензор.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает результат умножения матриц.
Пример использования
Пример
import torch
# Создать две матрицы
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# Умножение матриц
C = torch.linalg.matmul(A, B)
print("Shape of A:", A.shape)
print("Shape of B:", B.shape)
print("Shape of C:", C.shape)
# Создать две матрицы
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# Умножение матриц
C = torch.linalg.matmul(A, B)
print("Shape of A:", A.shape)
print("Shape of B:", B.shape)
print("Shape of C:", C.shape)
Результат вывода:
A 的形状: torch.Size([3, 4]) B 的形状: torch.Size([4, 5]) C 的形状: torch.Size([3, 5])
Другие расширения