Mojo function

iota

iota[dtype: DType, width: Int](offset: SIMD[dtype, 1] = __init__[__mlir_type.!pop.int_literal](0)) -> SIMD[dtype, width]

Creates a SIMD vector containing an increasing sequence, starting from offset.

Parameters:

dtype (DType): The dtype of the input and output SIMD vector.
width (Int): The width of the input and output SIMD vector.

Args:

offset (SIMD[dtype, 1]): The value to start the sequence at. Default is zero.

Returns:

An increasing sequence of values, starting from offset.

iota[dtype: DType, //](buff: UnsafePointer[SIMD[dtype, 1], address_space=address_space, alignment=alignment, origin=origin], len: Int, offset: Int = 0)

Fill the buffer with numbers ranging from offset to offset + len - 1, spaced by 1.

The function doesn't return anything, the buffer is updated inplace.

Parameters:

dtype (DType): DType of the underlying data.

Args:

buff (UnsafePointer[SIMD[dtype, 1], address_space=address_space, alignment=alignment, origin=origin]): The buffer to fill.
len (Int): The length of the buffer to fill.
offset (Int): The value to fill at index 0.

iota[dtype: DType, //](mut v: List[SIMD[dtype, 1], hint_trivial_type], offset: Int = 0)

Fill a list with consecutive numbers starting from the specified offset.

Parameters:

dtype (DType): DType of the underlying data.

Args:

v (List[SIMD[dtype, 1], hint_trivial_type]): The list to fill with numbers.
offset (Int): The starting value to fill at index 0.

iota(mut v: List[Int, hint_trivial_type], offset: Int = 0)

Fill a list with consecutive numbers starting from the specified offset.

Args:

v (List[Int, hint_trivial_type]): The list to fill with numbers.
offset (Int): The starting value to fill at index 0.