python - Tensordot 在 Tensorflow 中的使用

Question

在我的python Tensorflow代码中，我有一个尺寸为（201、25、25）的3d张量“a”和尺寸为（201、25）的2d张量“b”，我想计算产品a* b 其中，结果应在其第 i 行中包含使用 tensordot 方法计算的以下乘积。

tf.tensordot(a[i,:,:],b[i,:],1) // This product has dim of (25,) 有没有一种使用 tensordot 函数一次计算这个产品的好方法，我试过这样做，但我做不到。