当前标签：torch.nonzero

torch.nonzero()和torch.index_select()，筛选张量中符合某种条件的元素
Picassooo 2021-08-25 22:07 阅读:13092 评论:0 推荐:1