Skip to main content
Ctrl+K

jax.lax.expand_dims#

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

Insert any number of size 1 dimensions into an array.

Parameters:
  • array (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

  • dimensions (Sequence[int]) –

Return type:

Array

previous

jax.lax.exp

next

jax.lax.expm1