在我学会了如何使用之后einsum,我现在正试图了解它是如何np.tensordot工作的.
但是,我有点迷失,特别是关于参数的各种可能性axes.
要理解它,因为我从未练习过张量微积分,我使用以下示例:
A = np.random.randint(2, size=(2, 3, 5))
B = np.random.randint(2, size=(3, 2, 4))
Run Code Online (Sandbox Code Playgroud)
在这种情况下,有什么不同可能np.tensordot,你会如何手动计算?