Ich möchte Tensordot verwenden, um das Skalarprodukt einer bestimmten Abmessung von zwei Tensoren zu berechnen. Mögen

A ist ein Tensor, dessen Form (3, 4, 5) ist. B ist ein Tensor, dessen Form (3, 5) ist.

Ich möchte einen Punkt mit A's drittem und B's zweitem Dim machen und eine Ausgabe erhalten, deren Dims (3, 4) ist

Wie unten

for i in range(3):
    C[i] = dot(A[i], B[i])

Wie mache ich das mit Tensordot?