Функция torch.mm в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.mmявляется функцией в PyTorch, используемой для выполнения двумерного матричного умножения.

Определение функции

torch.mm(input, mat2, out)

Пример использования

Пример

import torch

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

# Умножение матриц
C = torch.mm(A, B)

print("Форма A:", A.shape)
print("Форма B:", B.shape)
print("Форма C:", C.shape)

Результат вывода:

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

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения