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