锦鲤

Pytorch 中文文档 https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/ Pytorch Tutorial https://pytorch.org/tutorials/index.html PyTorch官方教程中文版 http://www.pytorch123.com/ ...

dataloader读进来的数据标签是torch.Size([n])，需要和数据做torch.cat操作，要保持两者的维度一致，变成torch.Size([n, 1]) 代码举例import torch import numpy as np label = torch.randperm(5) print(label) # tensor([1, 3, 0, 4, 2]) print(label...