ones

nvtripy.ones(shape: nvtripy.types.ShapeLike, dtype: dtype = float32) → Tensor

Creates a Tensor of the specified shape and dtype with all elements set to 1.

Parameters:

• shape (nvtripy.types.ShapeLike) – The desired shape of the tensor.

• dtype (dtype) – [dtype=T1] Datatype of elements.

Returns:

[dtype=T1] A tensor of shape shape with all elements set to 1.

Return type:

Tensor

TYPE CONSTRAINTS:

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

Example

output = tp.ones([2, 3])

Local Variables

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

See also

ones_like(), full()