Trait dfdx::tensor_ops::Select1

pub trait Select1<T, const I: isize> {
    type Indices: Clone;

    fn select(self, indices: &Self::Indices) -> T;
}

Select values along a single axis I resulting in T. Equivalent to torch.select and torch.gather from pytorch.

Required Associated Types

type Indices: Clone

Required Methods

fn select(self, indices: &Self::Indices) -> T

Select sub elements using Self::Indices. The same element can be selected multiple times depending on Self::Indices.

Implementors

impl<const M: usize, H: Tape> Select1<Tensor0D<H>, -1> for Tensor1D<M, H>

type Indices = usize

impl<const M: usize, const N: usize, H: Tape> Select1<Tensor1D<M, H>, -1> for Tensor2D<M, N, H>

type Indices = [usize; M]

impl<const M: usize, const N: usize, H: Tape> Select1<Tensor1D<N, H>, 0> for Tensor2D<M, N, H>

type Indices = usize

impl<const M: usize, const N: usize, const O: usize, H: Tape> Select1<Tensor2D<M, N, H>, -1> for Tensor3D<M, N, O, H>

type Indices = [[usize; N]; M]

impl<const M: usize, const N: usize, const O: usize, H: Tape> Select1<Tensor2D<M, O, H>, 1> for Tensor3D<M, N, O, H>

type Indices = [usize; M]

impl<const M: usize, const N: usize, const O: usize, H: Tape> Select1<Tensor2D<N, O, H>, 0> for Tensor3D<M, N, O, H>

type Indices = usize

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Select1<Tensor3D<M, N, O, H>, -1> for Tensor4D<M, N, O, P, H>

type Indices = [[[usize; O]; N]; M]

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Select1<Tensor3D<M, N, P, H>, 2> for Tensor4D<M, N, O, P, H>

type Indices = [[usize; N]; M]

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Select1<Tensor3D<M, O, P, H>, 1> for Tensor4D<M, N, O, P, H>

type Indices = [usize; M]

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Select1<Tensor3D<N, O, P, H>, 0> for Tensor4D<M, N, O, P, H>

type Indices = usize

impl<const M: usize, const N: usize, const O: usize, const P: usize, const Z: usize, H: Tape> Select1<Tensor4D<M, N, O, Z, H>, -1> for Tensor4D<M, N, O, P, H>

type Indices = [[[[usize; Z]; O]; N]; M]

impl<const M: usize, const N: usize, const O: usize, const P: usize, const Z: usize, H: Tape> Select1<Tensor4D<M, N, Z, P, H>, 2> for Tensor4D<M, N, O, P, H>

type Indices = [[[usize; Z]; N]; M]

impl<const M: usize, const N: usize, const O: usize, const P: usize, const Z: usize, H: Tape> Select1<Tensor4D<M, Z, O, P, H>, 1> for Tensor4D<M, N, O, P, H>

type Indices = [[usize; Z]; M]

impl<const M: usize, const N: usize, const O: usize, const P: usize, const Z: usize, H: Tape> Select1<Tensor4D<Z, N, O, P, H>, 0> for Tensor4D<M, N, O, P, H>

type Indices = [usize; Z]

impl<const M: usize, const N: usize, const O: usize, const Z: usize, H: Tape> Select1<Tensor3D<M, N, Z, H>, -1> for Tensor3D<M, N, O, H>

type Indices = [[[usize; Z]; N]; M]

impl<const M: usize, const N: usize, const O: usize, const Z: usize, H: Tape> Select1<Tensor3D<M, Z, O, H>, 1> for Tensor3D<M, N, O, H>

type Indices = [[usize; Z]; M]

impl<const M: usize, const N: usize, const O: usize, const Z: usize, H: Tape> Select1<Tensor3D<Z, N, O, H>, 0> for Tensor3D<M, N, O, H>

type Indices = [usize; Z]

impl<const M: usize, const N: usize, const Z: usize, H: Tape> Select1<Tensor2D<M, Z, H>, -1> for Tensor2D<M, N, H>

type Indices = [[usize; Z]; M]

impl<const M: usize, const N: usize, const Z: usize, H: Tape> Select1<Tensor2D<Z, N, H>, 0> for Tensor2D<M, N, H>

type Indices = [usize; Z]

impl<const M: usize, const Z: usize, H: Tape> Select1<Tensor1D<Z, H>, -1> for Tensor1D<M, H>

type Indices = [usize; Z]