ToMode

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

Change input data to a target mode. For example, most transforms use HWC mode image, while the neural network might use CHW mode input tensor.

Parameters:

• mode – output mode of input. Default: “CHW”

• order – the same with VisionTransform