您可以简单地使用张量流的广播功能。
import tensorflow as tfx = tf.constant([[0, 1],[2, 3],[4, 5],[6, 7]], dtype=tf.float32)y = tf.constant([[0, 1],[2, 3]], dtype=tf.float32)x_ = tf.expand_dims(x, 0)y_ = tf.expand_dims(y, 1)z = tf.reshape(tf.add(x_, y_), [-1, 2])# or more succinctly z = tf.reshape(x[None] + y[:, None], [-1, 2])sess = tf.Session()sess.run(z)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)