... + 1) order = torch.sort(scores, descending=True)[1] #(?,) keep =[] while order.numel() > 0: if order.numel() == 1: keep.append(order.item()) break else: ...
確定! 回上一頁