Как умножить все строки на столбец одной матрицы в Pytorch в векторизации - PullRequest
0 голосов
/ 18 января 2019

Мне нужно умножить все строки матрицы на столбец, я думаю, с примером это будет понятнее:

матрица:

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

Мне нужна операция, которая возвращает:

28,80,162

Но я не могу найти ничего в документации и блогах, и другие вопросы SO связаны только с умножением матриц и точечным произведением, что не является тем, что мне нужно в данном случае, как это может быть достигнуто в векторизованном виде (вместо этого из на основе цикла)?

Например, это легко достичь в случае суммы, такой как:

the_matrix.sum(dim=0)

Но нет ничего подобного:

the_matrix.mul(dim=0)

1 Ответ

0 голосов
/ 18 января 2019

Я нашел решение, нет:

the_matrix.mul(dim=0)

Но есть:

he_matrix.prod(dim=0)

Что делает именно то, что нужно.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...