HeNormal

class vulkpy.nn.HeNormal

Bases: Initializer

He Normal Initializer

Note

Standard deviation \(\sigma\) is following;

\[\sigma = \sqrt{2/d_{\text{in}}}\]

Methods Summary

__call__(gpu, shape)

Initialize new parameters

Methods Documentation

__call__(gpu: GPU, shape: Iterable[int])

Initialize new parameters

Parameters:

• gpu (vulkpy.GPU) – GPU

• shape (iterable of ints) – Parameter shape

__init__(gpu: GPU, input_dim: int, *, seed: int | None = None)

Initialize He Normal Initializer

Parameters:

• gpu (vulkpy.GPU) – GPU

• input_dim (int) – Input dimension

• seed (int, optional) – Initial seed for PRNG