split¶
dragon.
split
(
inputs,
num_or_size_splits,
axis=0,
copy=True,
**kwargs
)[source]¶Split input into chunks along the given axis.
Either number or size of splits will be accepted:
x = dragon.constant([[1, 2], [3, 4], [5, 6]]) # Shape: (3, 2) -> (2, 2), (1, 2) print(dragon.split(x, num_or_size_splits=2)) # Shape: (3, 2) -> (1, 2), (2, 2) print(dragon.split(x, num_or_size_splits=(1, 2)))
axis
can be negative:x = dragon.constant([[1, 2], [3, 4], [5, 6]]) print(dragon.split(x, 2, axis=1)) print(dragon.split(x, 2, axis=-1)) # Equivalent
- Parameters:
- inputs (dragon.Tensor) – The input tensor.
- num_or_size_splits (Union[int, Sequence[int]]) – The number or size of chunks.
- axis (int, optional, default=0) – The axis to split.
- copy (bool, optional, default=True) – Copy or create the views of input.
- Returns:
Sequence[dragon.Tensor] – The output tensors.