torch.mm

编程入门行业动态更新时间:2024-10-23 07:23:12

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

执行矩阵input和mat2的矩阵乘法运算。

如果input是（n×m）张量，mat2是（m×p）张量，out将是（n x p）张量。

input（张量）–第一个要矩阵相乘的矩阵

mat2（张量）–第二个要矩阵相乘的矩阵

out（张量，可选）–输出张量。

>>> 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]])

更多推荐

torch.mm

本文发布于:2023-11-15 19:33:56，感谢您对本站的认可！

torch mm

评论列表（有 0 条评论）