torch.nonzero()和torch.index_select(),筛选张量中符合某种条件的元素 Picassooo 博客园(03) Save Image Image gallery: Torch.index_select