Я пытаюсь использовать torch.bmm, чтобы выполнить следующую матричную операцию,
Если матрица - это тензор M * N, партия - это тензор N * B, как я могу этого добиться? В каждой партии, matrix @ batch_i, который дает M и складывает размер пакета вместе, тензор вывода выглядит как M * B
Здесь есть два вопроса,
1. Для использования torch.bmm, кажется нужно, чтобы обе матрицы были пакетными, но мой первый ввод не
Размер партии должен быть первым измерением, а размер моей партии в конце
Я думаю, это тот же вопрос для Numpy пользователей