megengine.amp.convert_tensor_format

convert_tensor_format(x, inplace=True)[源代码]

将 NCHW 张量转换为 NHWC 张量。