megengine.amp.convert_tensor_format#

convert_tensor_format(x, inplace=True)[source]#

Convert NCHW Tensor to NHWC Tensor.