PyTorch torch.bmm 関数
Pytorch torch リファレンスマニュアルtorch.bmmPyTorch でバッチ行列乗算を実行するための関数です。
関数定義
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)
# バッチ行列乗算
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])
その他の拡張