torch 列最大值筛选满足条件的行或列（使用与或）
max_v= torch.max(scores_s, 1)[0]下面的代码是一样的： index_num = torch.arange(0, conf.size(0)) max_ids= torch.max(scores_s, 1)[1] scores_v=scores_s[index_num,max...
秀丽红牛
他的专栏
他的归档
热门文章
