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]