python-3.x - 如何以矢量化方式将pytorch中单个矩阵的所有行乘以列

Question

我需要将矩阵的所有行逐列相乘，我认为举个例子会更清楚：

矩阵是：

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

我需要一个返回的操作：

28,80,162

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

例如，对于 sum 的情况，这很容易实现，例如：

the_matrix.sum(dim=0)

但是没有类似的东西：

the_matrix.mul(dim=0)

标签： python-3.xmathdeep-learningpytorch

Answer 1

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

the_matrix.mul(dim=0)

但是有：

he_matrix.prod(dim=0)

这正是需要的。