模型开发接口
推理部署接口
API 相关说明
Calculates fan_in / fan_out value for given weight tensor, depending on given mode.
mode
See calculate_fan_in_and_fan_out for details.
calculate_fan_in_and_fan_out
tensor (Tensor) – weight tensor in NCHW format.
Tensor
NCHW
mode (str) – fan_in” or “fan_out”.
str
float
previous
megengine.module.init.calculate_fan_in_and_fan_out
next
megengine.module.init.xavier_uniform_