slice¶

dragon.slice(
  inputs,
  starts,
  sizes,
  **kwargs
)[source]¶

Select the elements according to the given sections.

Each section should be hinted by a pair of [start, start + size):

x = dragon.constant([[[0, 1, 2], [3, 4, 5]]])
print(dragon.slice(x, [0, 1, 2], [1, 1, 1]))  # [[[5]]]
print(x[0:1, 1:2:, 2:3])  # Equivalent

sizes accepts value -1 or 0:

x = dragon.constant([[[0, 1, 2], [3, 4, 5]]])
# Set ``0`` to squeeze dimensions with size 1
print(dragon.slice(x, [0, 1, 2], [0, 0, 0]))  # 5
# Set ``-1`` to take all the remained elements
print(dragon.slice(x, [0, 0, 0], [-1, -1, -1]))  # [[[0, 1, 2], [3, 4, 5]]]

Parameters:

• inputs (dragon.Tensor) – The input tensor.
• starts (Union[Sequence[int], dragon.Tensor]) – The start location for each dimension.
• sizes (Union[Sequence[int], dragon.Tensor]) – The number of elements sliced from start.

Returns:

dragon.Tensor – The output tensor.