Function burn::tensor::narrow

pub fn narrow<B, K>(
    tensor: <K as TensorKind<B>>::Primitive,
    dim: usize,
    start: usize,
    length: usize,
) -> <K as TensorKind<B>>::Primitive
where
    B: Backend,
    K: TensorKind<B> + BasicOps<B>,
Expand description

Returns a new tensor with the given dimension narrowed to the given range.

§Arguments

  • tensor - The tensor.
  • dim - The dimension along which the tensor will be narrowed.
  • start - The starting point of the given range.
  • length - The ending point of the given range.

§Panics

  • If the dimension is greater than the number of dimensions of the tensor.
  • If the given range exceeds the number of elements on the given dimension.

§Returns

A new tensor with the given dimension narrowed to the given range.