jax.lax.broadcasted_iota#

jax.lax.broadcasted_iota(dtype, shape, dimension)[source]#

Convenience wrapper around iota.

Parameters
Return type

Array