python - Numpy:将矩阵与 3d 张量相乘——建议

标签 python matrix numpy scipy linear-algebra

我有一个形状为 MxN 的矩阵 P 和一个形状为 KxNxR 的 3d 张量 T。我想将 PT 中的每个 NxR 矩阵相乘,得到 KxMxR 3d 张量。

P.dot(T).transpose(1,0,2) 给出了想要的结果。这个问题有没有更好的解决方案(即摆脱transpose)?这一定是一个很常见的操作,所以我认为,其他人已经找到了不同的方法,例如使用 tensordot(我试过但没能得到想要的结果)。意见/观点将不胜感激!

最佳答案

scipy.tensordot(P, T, axes=[1,1]).swapaxes(0,1)

关于python - Numpy:将矩阵与 3d 张量相乘——建议,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/4490961/

相关文章:

python - sympy 中矩阵的按列(或按行)运算

r - 在 R 中按组有效地复制矩阵行

python - 在循环中使用 numpy load 时内存溢出

组织矩阵以使邻居最接近的算法

python - 我想从txt文件中提取x和y值

python - 具有多个同时序列的 Keras 序列预测

python - 如何从 python 中的元素列表中删除特殊字符?

python - Python OpenCV-检测眼睛并保存

python - 添加列并选择总和最大的列

python - Graphviz:Python 中记录标签的边 (gv)