Function burn_tensor::narrow

pub fn narrow<B: Backend, const D: usize, K: TensorKind<B> + BasicOps<B>>(
    tensor: K::Primitive<D>,
    dim: usize,
    start: usize,
    length: usize
) -> K::Primitive<D>

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.