摘要：
1. 数据输入 pytorch 中计计算交叉熵损失函数时， 输入的正确 label 不能是 one-hot 格式。函数内部会自己处理成 one hot 格式。所以不需要输入 [ 0 0 0 0 1]，只需要输入 4 就行。 在经过 loss 的时候，CrossEntropyLoss 会自动为其编码为 阅读全文
摘要：
1. view( )函数 PyTorch 中的view( )函数相当于numpy中的resize( )函数，都是用来重构(或者调整)张量维度的，用法稍有不同。 >>> import torch >>> re = torch.tensor([1, 2, 3, 4, 5, 6]) >>> result 阅读全文
摘要：
TensorDataset TensorDataset可以用来对 tensor 进行打包，就好像 python 中的 zip 功能。该类通过每一个 tensor 的第一个维度进行索引。因此，该类中的 tensor 第一维度必须相等. 另外：TensorDataset 中的参数必须是 tensor i 阅读全文