megengine.amp.convert_tensor_format

convert_tensor_format(x, inplace=True)

Convert NCHW Tensor to NHWC Tensor.

上一页

megengine.amp.convert_module_format

下一页

megengine.amp.enabled