theano - use o produto tensordot de ponto computacional de dois tensores
Eu quero usar o tensordot para calcular o produto escalar de uma intensidade específica de dois tensores. Gostar:
A é um tensor, cuja forma é (3, 4, 5) B é um tensor, cuja forma é (3, 5)
Eu quero fazer um ponto usar a terceira dim de A e a segunda dim de B e obter uma saída cujas dims sejam (3, 4)
Como abaixo:
for i in range(3):
C[i] = dot(A[i], B[i])
Como fazê-lo por tensordot?