Noções básicas sobre tensordot
Depois que aprendi a usareinsum
, Agora estou tentando entender comonp.tensordot
trabalho.
No entanto, estou um pouco perdido, especialmente em relação às várias possibilidades para o parâmetroaxes
.
Para entendê-lo, como nunca pratiquei cálculo tensorial, use o seguinte exemplo:
A = np.random.randint(2, size=(2, 3, 5))
B = np.random.randint(2, size=(3, 2, 4))
Nesse caso, quais são as diferentes possíveisnp.tensordot
e como você o calcularia manualmente?