python - 沿任意轴相乘？

Question

如果我有一个形状为 [batch_size, height, width, depth] 的层的输出和另一个形状为 [depth] 的张量，我如何将第一个张量乘以第二个，这样沿 depth 方向的每个切片都乘以第二个张量中的相应值。也就是说，如果第二个张量是[4, 5, 6]，那么乘法是:

tensor1[:, :, :, 0] * 4
tensor1[:, :, :, 1] * 5
tensor1[:, :, :, 2] * 6

还有，不知道有没有这种乘法的名字去搜一下？谢谢!

Answer 1

这很简单。只需将两个张量相乘即可。例如:

import tensorflow as tf

tensor = tf.Variable(tf.ones([2, 2, 2, 3]))
depth = tf.constant([4, 5, 6], dtype=tf.float32)
result = tensor * depth

sess = tf.Session()
sess.run(tf.initialize_all_variables())
print(sess.run(result))