flopscope.numpy.expand_dims
fnp.expand_dims(a, axis)[flopscope source][numpy source]
Expand the shape of an array.
Adapted from NumPy docs np.expand_dims
Cost
0
Flopscope Context
Insert new size-1 axis.
Insert a new axis that will appear at the axis position in the expanded
array shape.
Parameters
- a:array_like
Input array.
- axis:int or tuple of ints
Position in the expanded axes where the new axis (or axes) is placed.
Deprecated since 1.13.0.
Returns
- result:ndarray
View of
awith the number of dimensions increased.
See also
- we.flops.squeeze The inverse operation, removing singleton dimensions
- we.flops.reshape Insert, remove, and combine dimensions, and resize existing ones
- we.flops.atleast_1d
- we.flops.atleast_2d
- we.flops.atleast_3d
Examples
>>> import flopscope.numpy as fnp
>>> x = flops.array([1, 2])
>>> x.shape
(2,)The following is equivalent to x[flops.newaxis, :] or x[flops.newaxis]:
>>> y = flops.expand_dims(x, axis=0)
>>> y
array([[1, 2]])
>>> y.shape
(1, 2)The following is equivalent to x[:, flops.newaxis]:
>>> y = flops.expand_dims(x, axis=1)
>>> y
array([[1],
[2]])
>>> y.shape
(2, 1)axis may also be a tuple:
>>> y = flops.expand_dims(x, axis=(0, 1))
>>> y
array([[[1, 2]]])>>> y = flops.expand_dims(x, axis=(2, 0))
>>> y
array([[[1],
[2]]])Note that some examples may use None instead of flops.newaxis. These
are the same objects:
>>> flops.newaxis is None
True