介绍

相当于 numpy布尔索引

语法

tf.boolean_mask(
    tensor,
    mask,
    axis=None,
    name='boolean_mask'
)

参数

• tensor：n维tensor，或者list
• mask：boolean张量，或者boolean型的list
• axis：从哪个轴覆盖，默认为0

例子

a=tf.constant([0,1,2,3])
# 下面代码也可用
# a=[0,1,2,3]
m=[True,False,True,False]
b=tf.boolean_mask(a,m)
print(b)

执行结果：

tf.Tensor([0 2], shape=(2,), dtype=int32)

原文出处：/show_1EF5DBlceZ8l.html