tensorflow 生成一个tensor的mask

最新推荐文章于 2024-01-11 21:42:01 发布

原创最新推荐文章于 2024-01-11 21:42:01 发布 · 3.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #tensor mask #tf.where

python tensorflow 专栏收录该内容

7 篇文章

订阅专栏

本文展示了一个使用TensorFlow进行条件操作的例子，具体为根据矩阵元素的正负值选择填充1或0，通过运行会话并打印结果，展示了TensorFlow中where函数的应用。

部署运行你感兴趣的模型镜像

import tensorflow as tf

a = tf.constant([[-1, 2, 3],
                 [2, -3, 4],
                 [5, 6, -7]], dtype=tf.float32)

d = tf.ones(shape=[3, 3], name='d')
e = tf.zeros(shape=[3, 3], name='e')

b = tf.where(a < 0, d, e)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(b))


# output
# [[1. 0. 0.]
#  [0. 1. 0.]
#  [0. 0. 1.]]