Skip to main content
Ctrl+K

jax.lax.iota#

jax.lax.iota(dtype, size)[source]#

Wraps XLA’s Iota operator.

Parameters:
  • dtype (Union[str, type[Any], dtype, SupportsDType]) –

  • size (int) –

Return type:

Array

previous

jax.lax.integer_pow

next

jax.lax.is_finite