Python 更改pytorch的批次、高度、宽度、alpha到批次、alpha、高度、宽度

我有一批像这样的图片

torch.Size（[10,512,512,3]）

我可以循环到图像，可以看到10个图像。但要把这批东西喂给pytorch，我必须把它转换成

火炬尺寸（[10,3,512,512]）

我尝试了很多方法，但都没有找到解决办法

---

我们怎样才能做到这一点呢？

使用

permute

：

import torch
x = torch.rand(10, 512, 512, 3)

y = x.permute(0, 3, 1, 2)

x.shape

：

torch.Size（[10,512,512,3]）

---

y.shape

：

torch.Size（[10,3,512,512]）

Lol我自己试了1-2个小时。谢谢你。同时，它也减慢了训练的速度。你能告诉我们如何加快训练速度吗？因为改变了训练顺序，它增加了额外的时间。如果你想减少训练时间。您需要明智地选择优化器和批量大小。是的，谢谢您的建议