Warning

This page was created from a pull request.

jax.lax.expand_dims

jax.lax.expand_dims(array, dimensions)

Insert any number of size 1 dimensions into an array.

Parameters

• array (Any) –

• dimensions (Tuple[int, …]) –

Return type

Any