PyTorch torch.linalg.matmul 関数
Pytorch torch リファレンスマニュアルtorch.linalg.matmulは、PyTorch の線形代数モジュールで行列乗算を実行する関数です。これはtorch.matmulのエイリアスであり、より統一された線形代数インターフェースを提供します。
関数定義
torch.linalg.matmul(input, other, out=None)
パラメータ:
input(Tensor): 最初の入力テンソル。other(Tensor): 2番目の入力テンソル。out(Tensor, オプション): 出力テンソル。
戻り値:
torch.Tensor: 行列乗算の結果を返します。
使用例
実例
import torch
# 2つの行列を作成
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# 行列乗算
C = torch.linalg.matmul(A, B)
print("A の形状:", A.shape)
print("B の形状:", B.shape)
print("C の形状:", C.shape)
# 2つの行列を作成
A = torch.randn(3, 4)
B = torch.randn(4, 5)
# 行列乗算
C = torch.linalg.matmul(A, B)
print("A の形状:", A.shape)
print("B の形状:", B.shape)
print("C の形状:", C.shape)
出力結果は次のとおりです：
A 的形状: torch.Size([3, 4]) B 的形状: torch.Size([4, 5]) C 的形状: torch.Size([3, 5])
その他の拡張