torch.where 另类用法

一般情况下torch.where得到传参为condition, a,b返回得到一个tensor

另一种用法为，a和b皆为两个布尔型两维tensor，shape相同
c=torch.where(a&b)（a&b就是a与b，其实是一个tensor）
则回返回两个tensor d,e，d是0维的index，e是1维的index，两个合在一起就是返回a和b中都是True的位置索引