Numpy: Умножение матрицы на трехмерный тензор - Предложение

У меня есть матрица P с формой MxN и трехмерным тензором ] T с формой KxNxR . Я хочу умножить P на каждую матрицу NxR в T , что приведет к трехмерному тензору KxMxR .

P.dot ( T) .transpose (1,0,2) дает желаемый результат. Есть ли более приятное решение (т.е. избавление от транспонирования ) этой проблемы? Это должно быть довольно распространенная операция, поэтому я предполагаю, что другие нашли разные подходы, например, используя tensordot (что я пробовал, но не смог получить желаемый результат). Мнения / мнения были бы весьма признательны!

13
задан Steve Tjoa 20 December 2010 в 18:46
поделиться