megengine.module.init.calculate_fan_in_and_fan_out

calculate_fan_in_and_fan_out(tensor)

对于给定的权重张量，计算fan_in / fan_out值。此函数假定输入张量存储格式为NCHW。

参数

tensor (Tensor) – NCHW 格式的权重张量。

返回类型

Tuple[float, float]

megengine.module.init.calculate_gain megengine.module.init.calculate_correct_fan