He Init Std

Problem

Return the He/Kaiming normal initialization standard deviation sqrt(2 / fan_in).

The signature is he_std(fan_in).

Hint

ReLU networks only need the fan-in correction factor of 2.

case 1

in: [2]

expected: 1

case 2

in: [100]

expected: 0.1414213562

case 3

in: [128]

expected: 0.125

case 4

in: [512]

expected: 0.0625

Starter codePython

import math
def he_std(fan_in):
    # Your code here
    pass

4 test cases · run your code against them in the editor