Я хочу использовать тензорот, чтобы вычислить скалярное произведение определенного dim из двух тензоров. Подобно:

A - тензор, форма которого (3, 4, 5) B - тензор, форма которого (3, 5)

Я хочу сделать точку, используя третье затемнение А и второе затемнение В, и получить вывод, значение которого равно (3, 4)

Как ниже:

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

Как это сделать с помощью tenordot?