我有两个维度分别为 (1, 100) 和 (784, 100) 的向量 A 和 B。我认为 A 会沿原始数据广播到与 B 相同的维度,但得到“维度必须相等”的错误消息。你能解释一下为什么吗?
广播具有相同等级(即2
)的矩阵似乎像它在 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)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句