tensorflow如何迭代列表张量

发布于 2022-09-06 06:51:08 字数 688 浏览 15 评论 0

clipboard.png

import tensorflow as tf

x = [0.8, 0.4]
y = [0.5, 0.5]

with tf.Session() as sess:
    init_op = tf.global_variables_initializer()
    sess.run(init_op)
    compare_results = tf.greater(x, y).eval()
    results = [1. if compare_result==True else 0. for compare_result in compare_results]
    # 转化为tensorflow张亮
    tf_results = tf.cast(results, tf.float32)
    print(sess.run(tf_results))

目前需求如下,x是我的模型输出向量,y是一个阈值,对于x向量中的每一个元素,如果大于阈值(此处为0.5)则置为1 否则置0 但是tensorflow好像没有对应的遍历列表的功能,所以我用上述代码实现了,不知道各位大牛有没有原生的优雅解决方法呢

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文