jax.lax.broadcasted_iotaΒΆ

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

Convenience wrapper around iota.

Parameters
Return type

Any