pytorch矩阵乘法

Posted 2021-02-21 picassooo

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch矩阵乘法相关的知识，希望对你有一定的参考价值。

torch.mm(mat1, mat2) performs a matrix multiplication of mat1 and mat2

a = torch.randint(0, 5, (2, 3))   # tensor([[3, 3, 2],
                                  #         [2, 2, 2]])
                                  
b = torch.randint(0, 6, (3, 1))   # tensor([[1],
                                  #         [4],
                                  #         [5]])

torch.mm(a, b)   # tensor([[11],
                 #        [17]])

以上是关于pytorch矩阵乘法的主要内容，如果未能解决你的问题，请参考以下文章