Primitive Iota

Creates a tensor with values increasing along the specified dimension.

Usage

nvl_iota(dim, dtype, shape)

Arguments

dim

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

dtype

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

shape

(integer())
Shape of the output tensor.

Value

tensorish