2020年3月24日
Pytorch取最小或最大的张量索引
摘要： Pytorch中根据索引取张量有很多方法，比如index_select和masked_select，和gt,ge等配合食用，但如果需要取出最小几个或最大几个张量的索引，则需要动手写一下 a = torch.tensor([2,3,1,5]) y,_ = torch.sort(a) mask = a. 阅读全文
posted @ 2020-03-24 20:48 嶙羽 阅读(6545) 评论(0) 推荐(0) 编辑