我有一个形状为的矩阵
使用
P
和一个形状为的三维张量T
。我希望将P
与T
中每个NxR
矩阵相乘,得到一个形状为的三维张量。使用
P.dot(T).transpose(1,0,2)
可以得到所需结果。是否有更好的方法(即去除transpose
)解决这个问题?由于这是一种常见的操作,因此我认为其他人已经找到了不同的方法,例如使用tensordot
(我尝试了但未能获得所需结果)。非常感谢您的意见/建议!
scipy.tensordot(P, T, axes=[1,1])
的结果看了好几个小时,因为维度被交换而感到绝望。没想到还有swapaxes
这个函数,谢谢! - osdf