当前位置：网站首页>TF中的条件语句；where()

TF中的条件语句；where()

2022-08-11 06:45:00 【phac123】

tf.where(条件语句, A, B)

条件语句为真返回A,为假返回B

import tensorflow as tf

a = tf.constant([1, 2, 3, 4, 99, 6])
b = tf.constant([2, 3, 4, 5, 6, 7])

c = tf.where(tf.greater(a, b), a, b)
print(c)

在这里插入图片描述

版权声明
本文为[phac123]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_42596275/article/details/126267200