文件名称：pytorch学习
文件大小：37KB
文件格式：PDF
更新时间：2024-01-27 18:09:06
c OR pytorch
1、Tensor 与tensorflow相同，在pytorch框架中，同样使用张量（tensor）来传输数据。我们可以对tensor的维度进行自定义，如果图像转化为tensor，那么一般来说我们需要将图像先转类型为numpy（此时数据维度为N x H x W X C，即数量x高x宽x通道数），然后再使用toTensor()将数据维度转变为N x C x H x W，因为在训练中，需要将通道数提到前面。同时，需要特别注意的是，在训练网络中，图像转为tensor后的数据类型一般会是float或是uint8（在处理遥感图像时，需要特别小心，避免直接将int16的类型直接转化造成最后结果与原图类型不一