- 当需要将 W x H x C 的图像格式转换为 C x W x H 时,在Tensor数据类型下可以使用 tensor.permute(1,2,0) 进行转换;而在 numpy.array 类型下,使用reshape往往会出现问题,此时需要使用 array.transpose进行维度变化,其余操作相同。
- Loss要同时具有 tensor.requires_grad 以及 tensor.grad_fn 两个属性,才能指导网络正确的反向传播。
2021-11-12 笔记 (图像维度变换、Loss反向传播属性)
最新推荐文章于 2024-09-02 21:33:07 发布