Tensorflow：将矩阵列乘以矢量元素并返回矩阵

Question

这是我想在Tensorflow中完成的工作。

我有2x2矩阵（可训练）

x_1 x_2
x_3 x_4

我有输入向量

a
b

我想将矩阵的每一列乘以向量的元素，然后取回下面的矩阵

ax_1 bx_2
ax_3 bx_4

我可以通过将矩阵的每一列声明为单独的变量来获得此结果，但是我想知道是否还有更优雅的解决方案。

Answer 1

感谢broadcasting，使用常规乘法运算符应该可以：

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]]