tf.greater(v1,v2)和tf.where(p1,p2,p3)

最新推荐文章于 2023-03-24 19:06:59 发布

明天也要加油鸭

最新推荐文章于 2023-03-24 19:06:59 发布

阅读量638

点赞数 4

CC 4.0 BY-SA版权

分类专栏： tensorflow 文章标签： tensorflow tf.greater() tf.where()

本文链接：https://blog.youkuaiyun.com/Dashi_Lu/article/details/88120181

tensorflow 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了TensorFlow中tf.greater和tf.where函数的使用方法。tf.greater用于比较两个张量的每个元素大小，而tf.where则根据条件选择不同张量的值。通过实例演示了如何在会话中运行这些操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.greater(v1,v2)

tf.where(p1,p2,p3)

tf.greater(v1,v2)

功能：比较两个输入张量的每一个元素的大小，并返回比较结果。

import tensorflow as tf

v1 = tf.constant([1,2,3,4])
v2 = tf.constant([4,3,2,1])

with tf.Session() as sess:
    print(sess.run(tf.greater(v1,v2)))   
    sess.close()

输出结果：[False False True True]

tf.where(p1,p2,p3)

参数p1是一个bool变量，也可以是一个表达式，返回值是True或者False。这个变量可以是一个也可以是一个列表，就是很多个True或者False组成的列表。

如果是True，返回p2，反之返回p3

#coding:utf-8

import tensorflow as tf

A = 100
B = tf.constant([1, 2, 3, 4])
C = tf.constant([1, 1, 1, 1])
D = tf.constant([0, 0, 0, 0])

with tf.Session() as sess:
    print(sess.run(tf.where(A > 1, 'A', 'B')))
    print(sess.run(tf.where(False, 'A', 'B')))
    print(sess.run(tf.where(B > 2, C, D)))

输出：b'A'
b'B'
[0 0 1 1]