Iota

Creates a tensor with values increasing along the specified dimension.

Usage

nv_iota(dim, dtype, shape)

Arguments

dim

(integer(1))
Dimension along which values increase.

dtype

(character(1) | stablehlo::TensorDataType)
Data type.

shape

(integer())
Shape.

Value

tensorish