tensorflow 生成一个tensor的mask

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

曲草

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

阅读量3.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： python tensorflow 文章标签： tensorflow tensor mask tf.where

本文链接：https://blog.youkuaiyun.com/HelloWorld1108/article/details/100922632

python tensorflow 专栏收录该内容

7 篇文章

订阅专栏

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

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

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.]]