ToMode

class ToMode(mode='CHW', *, order=None)

将输入数据变换到目标模式。例如，大多数变换使用HWC模式图像，而神经网络可能使用CHW模式的输入张量。

参数:

• mode – 输入的输出模式。默认情况下使用”CHW”模式。

• order – 与 VisionTransform 一致