megengine.data.transform.ToMode¶
-
class
ToMode(mode='CHW', *, order=None)[源代码]¶ 基类：
megengine.data.transform.vision.transform.VisionTransform将输入数据变换到目标模式。例如，大多数变换使用HWC模式图像，而神经网络可能使用CHW模式的输入张量。
- 参数
mode – 输入的输出模式。默认情况下使用”CHW”模式。
order – 与
VisionTransform一致
方法
__init__([mode, order])初始化方法。
apply(input)对单个输入数据实施数据增强。
apply_batch(inputs)对一批（batch）的输入数据实施数据增广。