Skip to content

init

Functions:

Name Description
scaled_kaiming_uniform_

Initialize a tensor with a Kaiming distribution scaled by initialization_scale.

scaled_kaiming_uniform_

scaled_kaiming_uniform_(
    t: Tensor, initialization_scale: float
) -> None

Initialize a tensor with a Kaiming distribution scaled by initialization_scale.

Parameters:

Name Type Description Default

t

Tensor

The tensor to initialize.

required

initialization_scale

float

The amount to scale the initialization by.

required