理解 tensordot

11 浏览
0 Comments

理解 tensordot

学会了如何使用einsum之后，我现在试图理解np.tensordot的工作原理。

然而，我有点迷茫，特别是对于参数axes的各种可能性。

为了理解它，因为我从未练习张量计算，我使用以下示例：

A = np.random.randint(2, size=(2, 3, 5))
B = np.random.randint(2, size=(3, 2, 4))

在这种情况下，有哪些不同的np.tensordot可能性，你会如何手动计算它？

0