Implement He (Kaiming) initialization for a weight matrix.
Signature: def he_init(fan_in: int, fan_out: int, seed: int = 0) -> np.ndarray
Return a weight matrix of shape (fan_out, fan_in) sampled from N(0, sqrt(2/fan_in)).
Use np.random.seed(seed) for reproducibility.
Math
Asked at
Output
Anything you print() in your code will show up here after you click Run.
Test Results