TensorFlow：如何对向量和张量进行点积？

Question

我在 a 的 TF 中有一个向量（表示一维张量）shape=(n,)：my_vector = tf.random.uniform(shape=[n])我有一个张量 a shape=(m, n)：my_tensor = tf.random.uniform(shape=[m, n])我想对my_tensorby的每一行进行点积，并从点积中my_vector获得一个向量。shape=(m,)我怎样才能做到这一点？我将它的元素想象得更清楚。

Answer 1

尝试这个

tf.tensordot(my_tensor, my_vector, 1)