将4维张量与1维张量相乘

Question

我有一个秩4张量与尺寸[batch_size, num_rows, num_cols, num_values]和秩1张量与尺寸[num_values] 我想计算第四列中的值与等级1 Tensor的点积，得出尺寸为[batch_size, num_rows, num_cols, 1]的等级4 Tensor，然后可以将其tf.squeeze转换为具有尺寸[batch_size, num_rows, num_cols] 。 有谁知道我该怎么做？

Answer 1

您可以使用tensordot或reduce_sum ：

a = tf.constant(np.random.rand(2, 3, 5, 7))
b = tf.constant(np.random.rand(7))

tf.tensordot(a, b, [-1, -1])  # <tf.Tensor 'Tensordot_1:0' shape=(2, 3, 5) dtype=float64>

tf.reduce_sum(a * b, axis=-1)  # <tf.Tensor 'Tensordot_1:0' shape=(2, 3, 5) dtype=float64>