Primitive Iota

Creates a tensor with values increasing along the specified dimension.

Usage

nvl_iota(dim, dtype, shape, start = 1L, ambiguous = FALSE)

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.

start

(integer(1))
Starting value.

ambiguous

(logical(1))
Whether the type is ambiguous. Ambiguous types usually arise from R literals (e.g., 1L, 1.0) and follow special promotion rules. See the vignette("type-promotion") for more details.

Value

tensorish
Has the given dtype and shape.

Implemented Rules

• stablehlo

StableHLO

Lowers to stablehlo::hlo_iota().

See also

nv_iota()

Examples

jit_eval(nvl_iota(dim = 1L, dtype = "i32", shape = 5L))
#> AnvilTensor
#>  1
#>  2
#>  3
#>  4
#>  5
#> [ CPUi32{5} ]