tf.boolean_mask 布尔掩码（布尔索引） 作者：马育民 • 2020-06-01 23:14 • 阅读：10101 # 介绍 通过bool值对tensor进行索引 # 语法 ``` tf.boolean_mask( tensor, mask, axis=None, name='boolean_mask' ) ``` ##### 参数 - tensor：n维tensor - mask：k维 bool型tensor，k <= n，k必须是静态值 - axis：轴，0维 int型tensor，默认值为0，表示按照某个轴进行掩盖（即索引），K + axis <= N ##### 返回值 (N-K+1)维，根据 bool 掩码 索引的值 # 例子 ``` import tensorflow as tf tens=tf.constant([1,2,3,4,5]) res=tf.boolean_mask(tens, [ True,True,True,False,False]) print(res) ``` 执行结果： ``` tf.Tensor([1 2 3], shape=(3,), dtype=int32) ``` 按照0轴，用bool列表对tens进行掩盖，被True掩盖的取出，被False掩盖的丢弃，所以返回值保留1，2，3 原文出处：http://malaoshi.top/show_1EF5dAquWY0V.html