mixmo.utils.torchutils._calculate_fan_in_and_fan_out

mixmo.utils.torchutils._calculate_fan_in_and_fan_out(tensor)[source]

Compute the minimal input and output sizes for the weight tensor