torch:matmul

(torch:matmul a b)

torch.matmul のランク規則に従う微分可能な行列積です。ベクトル同士は内積 (スカラーテンソル)、行列とベクトルは通常の積、どちらかがランク 3 以上ならバッチ積 (linalg:matmul: 末尾 2 軸が行列で先頭の軸はブロードキャスト) になります。勾配は両オペランドに流れ (行列の場合 g . b^T と a^T . g)、バッチ軸は他のブロードキャスト随伴と同様に合計で縮約されます。