Trait tensor_rs::tensor::index_slicing::IndexSlicing

source · [−]
pub trait IndexSlicing where
    Self: Sized, {

Show 13 methods    fn cat(&self, tensors: &[&Self], dim: usize) -> Self;

    fn chunk(&self, chunks: usize, dim: usize) -> Vec<Self>;

    fn gather(&self, dim: usize, index: &Self) -> Self;

    fn index_select(&self, dim: usize, index: &Self) -> Self;

    fn reshape(&self, new_shape: &[usize]) -> Self;

    fn split(&self, sections: &[usize], dim: usize) -> Vec<Self>;

    fn squeeze(&self, dim: Option<usize>) -> Self;

    fn stack(tensors: &[&Self], dim: usize) -> Self;

    fn take(&self, index: &[usize]) -> Self;

    fn permute(&self, dims: &[usize]) -> Self;

    fn unsqueeze(&self, dim: usize) -> Self;

    fn conditional_select(&self, x: &Self, y: &Self) -> Self;

    fn repeat(&self, sizes: &[usize]) -> Self;

}

Required methods

Implementors