repeat

dragon.repeat(
  inputs,
  axis=None,
  repeats=1,
  **kwargs
)[source]

Repeat the elements along the given axis.

If axis is None, flattened results will be returned.

Parameters:
  • inputs (dragon.Tensor) – The input tensor.
  • axis (int, optional) – The axis to repeat.
  • repeats (Union[int, dragon.Tensor], optional, default=1) – The magnitude of repeating.
Returns:

dragon.Tensor – The output tensor.