ToTensor

2020-03-13  本文已影响0人  elementmaster

ToTensor实例将图像数据从uint8格式变换成32位浮点数格式

ToTensor实例还将图像通道从最后一维移到最前一维来方便之后介绍的卷积神经网络计算。通过数据集的transform_first函数,我们将ToTensor的变换应用在每个数据样本(图像和标签)的第一个元素,即图像之上。

上一篇 下一篇

猜你喜欢

热点阅读