torch.topk 功能:找出前k大的数据,及其索引号 input:张量 k:决定选取k个值,k=1是为top-1 dim:索引维度 返回: Tensor:前k大的值 LongTensor:前k大的值所在的位置 torch.topk(input, k, dim=None,largest=True,sorted=True,out=None ) torchvision.utils.make_grid 功能:制作网格图像 tensor:图像数据,BCHW形式 nrow:行数(列数自动计算) padd…