Матрица Pytorch с поэлементным умножением

Question

Допустим, у меня есть два тензора A формы: [32, 512] и B формы: [32, 512],
И я хочу сделать поэлементное умножение между векторами матрицы, чтобы получить новая матрица формы: [32, 1] (первая строка A с первой строкой B, а вторая строка A со второй строкой B и т. д.), текущие методы, которые я пробовал, просто мульти значения матрицы, а не строк и дает матрицу формы: [32, 512].
Спасибо!

Answer 1

Полагаю, это эквивалентно тому, что вы ищете, сначала умножение матриц, а затем сумма по каждой строке. [ассоциативный закон]

import torch

a = torch.zeros(31,512)
b = torch.zeros(31,512)

c = torch.sum(torch.mm(a,torch.t(b)), dim=1).unsqueeze(-1)
print(c.shape)

Out:

torch.Size([31, 1])