Python PyTorch mm用法及代碼示例

本文簡要介紹python語言中 torch.mm 的用法。

用法:

torch.mm(input, mat2, *, out=None) → Tensor

參數：

• input(Tensor) -要矩陣相乘的第一個矩陣

• mat2(Tensor) -要矩陣相乘的第二個矩陣

關鍵字參數：

out(Tensor,可選的) -輸出張量。

執行矩陣 input 和 mat2 的矩陣乘法。

如果input是(n \times m)張量，mat2是(m \times p)張量，out將是(n \times p)張量。

注意

此函數不廣播。對於廣播矩陣產品，請參閱 torch.matmul() 。

支持跨步和稀疏二維張量作為輸入，對於跨步輸入的自動分級。

該運算符支持 TensorFloat32。

例子：

>>> mat1 = torch.randn(2, 3)
>>> mat2 = torch.randn(3, 3)
>>> torch.mm(mat1, mat2)
tensor([[ 0.4851,  0.5037, -0.3633],
        [-0.0760, -3.6705,  2.4784]])