tf.where()返回一个布尔张量中真值的位置。对于非布尔型张量,非0的元素都判为True。
返回的是二维张量,第一个维度的数量,即行数表明有多少个为True值,同一行中的数据代表该True值在原张量中的位置。
tf.where(input_tensor,a,b)
作用:如果input_tensor对应位置为True,则张量a中的该位置处元素保留,反之,由张量b中相应位置的元素代替。
tf.where()返回一个布尔张量中真值的位置。对于非布尔型张量,非0的元素都判为True。
返回的是二维张量,第一个维度的数量,即行数表明有多少个为True值,同一行中的数据代表该True值在原张量中的位置。
tf.where(input_tensor,a,b)
作用:如果input_tensor对应位置为True,则张量a中的该位置处元素保留,反之,由张量b中相应位置的元素代替。