mixmo.utils.torchutils.
_calculate_correct_fan
Return the minimal input or output sizes for the weight tensor depending on which is needed