tensorflow如何迭代列表张量
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 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论