tf.where(原tf.select) 和tf.greater的使用

import tensorflow as tf

v1 = tf.constant([1.0, 2.0, 3.0, 4.0])
v2 = tf.constant([4.0, 3.0, 2.0, 1.0])

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print sess.run(tf.greater(v1, v2))
    # [False False  True  True]

    print sess.run(tf.where(tf.greater(v1, v2), v1, v2))
    # [ 4.  3.  3.  4.]

 

snake

作者: snake

我们需要为这个社会做一点贡献,失去了才懂得去珍惜。

发表评论

电子邮件地址不会被公开。 必填项已用*标注