Size ([2, 3, 28, 28]) 含义,选择第一、二两张图片print(a[:2, :1].shape) # torch.Size([2, 1, 28, 28]) 含义,取前两张图片第一个通道上的 ...
確定! 回上一頁