repeat¶
dragon.
repeat
(
inputs,
axis=None,
repeats=1,
**kwargs
)[source]¶Repeat the elements along the given axis.
Examples:
x = dragon.constant([[1, 2], [3, 4]]) # A negative axis is the last-k axis print(dragon.repeat(x, axis=1, repeats=2)) # [[1, 1, 2, 2], [3, 3, 4, 4]] print(dragon.repeat(x, axis=-1, repeats=2)) # Equivalent # If axis is None, repeat a flattened input print(dragon.repeat(x, repeats=2)) # [1, 1, 2, 2, 3, 3, 4, 4]
- Parameters:
- inputs (dragon.Tensor) – The input tensor.
- axis (int, optional) – The axis to repeat.
- repeats (Union[int, dragon.Tensor], optional, default=1) – The repeat size.
- Returns:
dragon.Tensor – The output tensor.