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

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

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

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

torch.bmm(input, mat2, out)

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

Пример

import torch

# Пакетное матричное умножение
batch_a = torch.randn(10, 3, 4)
batch_b = torch.randn(10, 4, 5)

result = torch.bmm(batch_a, batch_b)

print("Форма результата батча:", result.shape)

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

批量结果形状: torch.Size([10, 3, 5])

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

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