如何以向量化方式在pytorch中将所有行乘以单个矩阵的列

Question

我需要将矩阵的所有行乘以列，我想举一个例子会更清楚：

矩阵是：

access="hasAnyRole('ROLE_ADMIN','ROLE_USER')"

我需要返回以下操作：

1,2,3 
4,5,6 
7,8,9

但是我在文档和博客中找不到任何内容，其他问题仅与矩阵乘法和点积相关，在这种情况下，这不是我所需要的，如何以矢量化方式实现（相反基于for循环）？

例如，对于总和这样的情况，这很容易实现：

28,80,162

但是没有类似的东西

the_matrix.sum(dim=0)

Answer 1

我找到了解决方案，没有了：

the_matrix.mul(dim=0)

但是有：

he_matrix.prod(dim=0)

到底是做什么的