tensorflow的boolean_mask函数

在mask中定义true,保留与其进行运算的tensor里的部分内容,相当于投影的功能。

mask与tensor的维度可以不相同的,但是对应的长度一定要相同,也就是要有一一对应的部分;

结果的维度 = tensor维度 - mask维度 + 1

以下是参考连接的例子,便于理解:

 

 

 

posted @ 2019-09-06 11:51  you-wh  阅读(1187)  评论(0编辑  收藏  举报
Fork me on GitHub