评估张量流中两个张量的行的所有对组合

评估张量流中两个张量的行的所有对组合,第1张

评估张量流中两个张量的行的所有对组合

您可以简单地使用张量流的广播功能。

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)


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5648027.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存