megengine.module.init.calculate_fan_in_and_fan_out

calculate_fan_in_and_fan_out(tensor)[源代码]

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

注解

MegEngine 中 group conv2d kernel 的形状为 (G, O/G, I/G, K, K)。该函数默认用 fan_out = O/G * K * K 进行计算，但在 PyTorch 中用``fan_out = O * K * K``。

参数

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

返回类型

Tuple[float, float]

上一页

megengine.module.init.calculate_gain

下一页

megengine.module.init.calculate_correct_fan