в пыторх.Я хочу умножить каждый вектор матрицы A на каждый вектор матрицы B:
A = M x N B = L x N result = (M x L) x N
Я попробовал это, и это работает:
torch.einsum('bj,aj->baj', input_unfolded, self.weights)
Вы можете создать любой шаблон умножения, используя этот подход.
Попробуйте:
result = A[:, None, :] * B[None, ...]