nn.init.constant¶

lucid.nn.init.constant(tensor: Tensor, val: int | float | complex) → None¶

The constant function fills the input tensor with a constant value.

Function Signature¶

def constant(tensor: Tensor, val: _Scalar) -> None

Parameters¶

tensor (Tensor): The tensor to be initialized.
val (_Scalar): The constant value to fill the tensor with.

Returns¶

None: The function modifies the tensor in-place with the constant value.

Examples¶

>>> import lucid
>>> from lucid.nn.init import constant
>>> tensor = lucid.zeros((3, 3))
>>> constant(tensor, val=5)
>>> print(tensor)
Tensor([[ 5,  5,  5],
        [ 5,  5,  5],
        [ 5,  5,  5]], requires_grad=False)