Warning

This page was created from a pull request.

jax.lax.broadcasted_iota¶

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

Convenience wrapper around iota.

Parameters

dtype (Any) –
shape (Sequence[int]) –
dimension (int) –

Return type