Qufang

摘要： 1. torch中的索引矩阵 torch中有很多场景下都会生成索引矩阵,索引矩阵的shape和tensor的shape是相同的 a_tensor, a_index = torch.topk(a, dim=1) # or a_tensor, a_index = torch.sort(a, dim=1) 阅读全文