ones_like¶

nvtripy.ones_like(input: Tensor, dtype: dtype | None = None) → Tensor[source]¶

Creates a tensor with all elements set to 1 of the same shape as the input tensor.

Parameters:

input (Tensor) – [dtype=T1] The input tensor.
dtype (dtype | None) – [dtype=T2] Datatype of elements. If set to None, the datatype of the input tensor is used.

Returns:

[dtype=T2] A tensor of the same shape as the input with all elements set to 1.

Return type:

Tensor

TYPE CONSTRAINTS:

T1: bool, float8, float16, bfloat16, float32, int4, int8, int32, int64
T2: bool, float8, float16, bfloat16, float32, int4, int8, int32, int64

Example

input = tp.zeros([2, 3], dtype=tp.float32)
output = tp.ones_like(input)

Local Variables¶

>>> input
tensor(
    [[0.0000, 0.0000, 0.0000],
     [0.0000, 0.0000, 0.0000]], 
    dtype=float32, loc=gpu:0, shape=(2, 3))

>>> output
tensor(
    [[1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000]], 
    dtype=float32, loc=gpu:0, shape=(2, 3))