Skip to main content

Ctrl+K

Site Navigation

Beginner’s guide
User’s Guide
API References
开发者指南
论坛
- 官网

GitHub
Bilibili

Site Navigation

Beginner’s guide
User’s Guide
API References
开发者指南
论坛
- 官网

GitHub
Bilibili

Section Navigation

模型开发接口

megengine
megengine.data
megengine.functional
megengine.module
megengine.module.init
megengine.autodiff
megengine.optimizer
megengine.jit
megengine.amp
megengine.dtr
megengine.distributed
megengine.quantization
megengine.traced_module
megengine.hub
megengine.random
megengine.config

推理部署接口

lite/tensor.h
lite/network.h
lite/global.h
lite/common_enum_c.h
megenginelite.tensor
megenginelite.network
megenginelite.global_setting
megenginelite.struct
megenginelite.utils

API 相关说明

与其它框架 API 进行对比
Deprecated APIs
环境变量（Env）设置

API References
megengine.amp
megengine.am...

megengine.amp.convert_module_format#

convert_module_format(module, inplace=True)[source]#: Convert NCHW Module to NHWC Module.

previous

autocast

next

megengine.amp.convert_tensor_format