如何执行具有不同维度的两个向量的逐元素乘法

Question

我有两个向量A＆amp; B具有尺寸（1,100）和（784,100）。我认为A将沿着原始广播到与B相同的维度，但是得到的错误是＆＃34;维度必须相等＆＃34;。你能解释一下原因吗？

Answer 1

广播具有相同等级的矩阵（即2）似乎可以作为it says on the tin：

import tensorflow as tf
tf.__version__
# 1.3.0

A = tf.constant([[1, 2], [3, 4], [5, 6]], dtype=tf.float32)
B = tf.constant([[1, -1]], dtype=tf.float32)

sess = tf.Session()
sess.run(A * B)
# array([[ 1., -2.],
#        [ 3., -4.],
#        [ 5., -6.]], dtype=float32)

sess.run(tf.multiply(A, B))
# array([[ 1., -2.],
#        [ 3., -4.],
#        [ 5., -6.]], dtype=float32)