- jax.numpy.expand_dims(a, axis)[source]#
Expand the shape of an array.
LAX-backend implementation of
Original docstring below.
Insert a new axis that will appear at the axis position in the expanded array shape.
a (array_like) – Input array.
Position in the expanded axes where the new axis (or axes) is placed.
Deprecated since version 1.13.0: Passing an axis where
axis > a.ndimwill be treated as
axis == a.ndim, and passing
axis < -a.ndim - 1will be treated as
axis == 0. This behavior is deprecated.
Changed in version 1.18.0: A tuple of axes is now supported. Out of range axes as described above are now forbidden and raise an AxisError.
result – View of a with the number of dimensions increased.
- Return type: