megengine.module.init.calculate_correct_fan

calculate_correct_fan(tensor, mode)[源代码]

计算对于给定的权重张量fan_in或fan_out值，这取决于给定的 mode。

请参阅 calculate_fan_in_and_fan_out 了解详情。

参数

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

  • mode (str) – fan_in” or “fan_out”.

返回类型

float

上一页

megengine.module.init.calculate_fan_in_and_fan_out

下一页

megengine.module.init.xavier_uniform_