mars.tensor.repeat#

mars.tensor.repeat(a, repeats, axis=None)[source]#

Repeat elements of a tensor.

Parameters

  • a (array_like) – Input tensor.

  • repeats (int or tensor of ints) – The number of repetitions for each element. repeats is broadcasted to fit the shape of the given axis.

  • axis (int, optional) – The axis along which to repeat values. By default, use the flattened input tensor, and return a flat output tensor.

Returns

repeated_tensor – Output array which has the same shape as a, except along the given axis.

Return type

Tensor

See also

tile

Tile a tensor.

Examples

>>> import mars.tensor as mt

>>> mt.repeat(3, 4).execute()
array([3, 3, 3, 3])
>>> x = mt.array([[1,2],[3,4]])
>>> mt.repeat(x, 2).execute()
array([1, 1, 2, 2, 3, 3, 4, 4])
>>> mt.repeat(x, 3, axis=1).execute()
array([[1, 1, 1, 2, 2, 2],
       [3, 3, 3, 4, 4, 4]])
>>> mt.repeat(x, [1, 2], axis=0).execute()
array([[1, 2],
       [3, 4],
       [3, 4]])