mars.tensor.expand_dims

mars.tensor.expand_dims(a, axis)

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

参见

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

实际案例

>>> 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]:

>>> 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:

>>> mt.newaxis is None
True

上一页

mars.tensor.broadcast_arrays

下一页

mars.tensor.squeeze