TensorFlow：如何点积向量和张量？
TensorFlow: how to dot product a vector and a tensor?
我在 shape=(n,):
的 TF 中有一个向量（表示一维张量）
my_vector = tf.random.uniform(shape=[n])
我有一个张量 shape=(m, n):
my_tensor = tf.random.uniform(shape=[m, n])
我想用 my_vector 对 my_tensor 的每一行进行点积，并从点积中获得一个 shape=(m,) 向量。我该怎么做？
我把它的元素想象得更清楚了。
试试这个
tf.tensordot(my_tensor, my_vector, 1)
我在 shape=(n,):
my_vector = tf.random.uniform(shape=[n])
我有一个张量 shape=(m, n):
my_tensor = tf.random.uniform(shape=[m, n])
我想用 my_vector 对 my_tensor 的每一行进行点积，并从点积中获得一个 shape=(m,) 向量。我该怎么做？
我把它的元素想象得更清楚了。
试试这个
tf.tensordot(my_tensor, my_vector, 1)