jax.lax.expand_dims

Warning

This page was created from a pull request (#9655).

jax.lax.expand_dims¶

jax.lax.expand_dims(array, dimensions)[source]¶

Insert any number of size 1 dimensions into an array.

Parameters

array (Any) –
dimensions (Sequence[int]) –

Return type

previous

jax.lax.exp

next

jax.lax.expm1