diagonal#

torchdyno.models.initializers.diagonal(shape, min_val=-1, max_val=1, dtype=torch.float32)[source]#

Diagonal random tensor.

Parameters:

  • shape (int) – shape of the tensor.

  • min_val (float, optional) – minimum value for uniform initialization. Defaults to -1.

  • max_val (float, optional) – maximum value for uniform initialization. Defaults to 1.

  • dtype (dtype)

Returns:

initialized tensor.

Return type:

torch.Tensor