当前标签：torch.index_select

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