narrow¶

dragon.vm.torch.narrow(
  input,
  dimension,
  start,
  length
)[source]¶

Return a narrowed tensor of input.

Parameters:

• input (torch.Tensor) – The input tensor.
• dimension (int) – The dimension to slice.
• start (int) – The starting position.
• length (int) – The distance to the ending position.

Returns:

dragon.vm.torch.Tensor – The output tensor.