如何解决使用 Tensordot 复制矩阵-矩阵乘法
在 PyKeops
package 中,没有可用的矩阵-矩阵乘法公式。相反,他们实现了类似于 numpy.tensordot
的东西。我有两个大小为 A,B
和 m x n
的矩阵 n x n
。有没有办法使用 A @ B
复制 numpy.tensordot
。
import numpy as np
m,n = 10,20
A = np.random.random((m,n))
B = np.random.random((n,n))
result1 = A @ B
result2 = np.tensordot(A,B,(1,1))
print(result1 == result2)
# =======================
# array([[False,False,# ....
解决方法
好的,我找到了一个答案,我们可以使用给定 tensordot
的以下参数来解决这个问题:
import numpy as np
m,n = 10,20
A = np.random.random((m,n))
B = np.random.random((n,n))
result1 = A @ B
result2 = np.tensordot(A,B,[(1,),(0,)])
print(result1 == result2)
# =======================
# array([[True,True,# ....
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。