gpt4 book ai didi

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

转载 作者:太空狗 更新时间:2023-10-29 17:33:30 27 4
gpt4 key购买 nike

我有一个形状为 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/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com