pytorch中的手动设置图片标签的方式

如果想自己手动设置一些标签，可以使用下面的方式：

import torch

label = torch.tensor([2,0,4,1,2]).long()

注意，一定要在后面加上.long()才行，因为标签是这个long型的数据