умножение вектора в pytorch - PullRequest
0 голосов
/ 06 мая 2019

в пыторх.Я хочу умножить каждый вектор матрицы A на каждый вектор матрицы B:

A      = M x N
B      = L x N
result = (M x L) x N

Ответы [ 2 ]

0 голосов
/ 07 мая 2019

Я попробовал это, и это работает:

torch.einsum('bj,aj->baj', input_unfolded, self.weights)

Вы можете создать любой шаблон умножения, используя этот подход.

0 голосов
/ 06 мая 2019

Попробуйте:

result = A[:, None, :] * B[None, ...]
...