mars.tensor.
expand_dims
Expand the shape of a tensor.
Insert a new axis that will appear at the axis position in the expanded array shape.
a (array_like) – Input tensor.
axis (int) – Position in the expanded axes where the new axis is placed.
res – Output tensor. The number of dimensions is one greater than that of the input tensor.
Tensor
See also
squeeze
The inverse operation, removing singleton dimensions
reshape
Insert, remove, and combine dimensions, and resize existing ones
doc.indexing, atleast_1d, atleast_2d, atleast_3d
doc.indexing
atleast_1d
atleast_2d
atleast_3d
Examples
>>> import mars.tensor as mt
>>> x = mt.array([1,2]) >>> x.shape (2,)
The following is equivalent to x[mt.newaxis,:] or x[mt.newaxis]:
x[mt.newaxis,:]
x[mt.newaxis]
>>> y = mt.expand_dims(x, axis=0) >>> y.execute() array([[1, 2]]) >>> y.shape (1, 2)
>>> y = mt.expand_dims(x, axis=1) # Equivalent to x[:,mt.newaxis] >>> y.execute() array([[1], [2]]) >>> y.shape (2, 1)
Note that some examples may use None instead of np.newaxis. These are the same objects:
None
np.newaxis
>>> mt.newaxis is None True