Warning

This page was created from a pull request.

jax.lax.iota

jax.lax.iota(dtype, size)

Wraps XLA’s Iota operator.

Parameters

• dtype (Any) –

• size (int) –

Return type

Any