摘要：
1. 数据输入 pytorch 中计计算交叉熵损失函数时， 输入的正确 label 不能是 one-hot 格式。函数内部会自己处理成 one hot 格式。所以不需要输入 [ 0 0 0 0 1]，只需要输入 4 就行。 在经过 loss 的时候，CrossEntropyLoss 会自动为其编码为 阅读全文
posted @ 2022-11-10 14:45
wangssd
阅读(321)
评论(0)
推荐(0)
摘要：
tensor张量大小： 四维图片数据：(Batch, Channels, Height, Width) 三维序列数据: (sequence_length, batch_size, input_size) 1. view( )函数 PyTorch 中的view( )函数相当于numpy中的resize 阅读全文
posted @ 2022-11-10 09:49
wangssd
阅读(262)
评论(0)
推荐(0)
摘要：
TensorDataset TensorDataset可以用来对 tensor 进行打包，就好像 python 中的 zip 功能。该类通过每一个 tensor 的第一个维度进行索引。因此，该类中的 tensor 第一维度必须相等. 另外：TensorDataset 中的参数必须是 tensor i 阅读全文
posted @ 2022-11-10 09:27
wangssd
阅读(229)
评论(0)
推荐(0)
浙公网安备 33010602011771号