Тензор потока: умножить столбцы матрицы на элементы вектора и вернуть матрицу обратно

Question

Вот что я хотел бы достичь в Tensorflow.

У меня есть матрица 2x2 (обучаемая)

x_1 x_2
x_3 x_4

, и у меня есть входной вектор

a
b

Я хотел бы умножить каждый столбец матрицы на элемент вектора ивернуть следующую матрицу

ax_1 bx_2
ax_3 bx_4

Я могу получить этот результат, объявив каждый столбец матрицы как отдельную переменную, но мне интересно, есть ли более элегантное решение.

Answer 1

Благодаря трансляции , у вас все будет в порядке, используя обычный оператор умножения:

import tensorflow as tf

x = tf.constant([[3, 5], [7, 11]], dtype=tf.int32)
a = tf.constant([4, 8], dtype=tf.int32)
y = x * a

with tf.Session() as sess:
    print(sess.run(y))  # Result: [[12, 40], [28, 88]]