TensorFlow: как скалярно произведение вектора и тензора?

У меня есть вектор (что означает одномерный тензор) в TF shape=(n,):

my_vector = tf.random.uniform(shape=[n])

И у меня есть тензор shape=(m, n):

my_tensor = tf.random.uniform(shape=[m, n])

Я хочу произвести точечное произведение каждой строки my_tensor на my_vector и получить вектор shape=(m,) из точечных произведений. Как я могу это сделать?

Я визуализировал его элементы, чтобы они были более четкими.

введите описание изображения здесь


person Peyman    schedule 25.10.2020    source источник


Ответы (1)


Попробуй это

tf.tensordot(my_tensor, my_vector, 1)
person dimay    schedule 25.10.2020