Skip to content

Fix _compute_fans when fan_in_axes is specified and len(shape) is 2. #2155

Fix _compute_fans when fan_in_axes is specified and len(shape) is 2.

Fix _compute_fans when fan_in_axes is specified and len(shape) is 2. #2155