Trait dfdx::tensor_ops::PermuteTo

pub trait PermuteTo<T, Axes> {
    fn permute(self) -> T;
}

Permutes self into T with the new order of axes specified via Axes.

Required Methods

fn permute(self) -> T

Permutes the tensor

Examples

let _: Tensor2D<3, 2> = Tensor2D::<2, 3>::zeros().permute();
let _: Tensor3D<3, 4, 2> = Tensor3D::<2, 3, 4>::zeros().permute();
let _: Tensor4D<3, 4, 5, 2> = Tensor4D::<2, 3, 4, 5>::zeros().permute();

Implementors

impl<const M: usize, const N: usize, H: Tape> PermuteTo<Tensor2D<M, N, H>, (Axis<0>, Axis<1>)> for Tensor2D<M, N, H>

impl<const M: usize, const N: usize, H: Tape> PermuteTo<Tensor2D<N, M, H>, (Axis<1>, Axis<0>)> for Tensor2D<M, N, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> PermuteTo<Tensor3D<M, N, O, H>, (Axis<0>, Axis<1>, Axis<2>)> for Tensor3D<M, N, O, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> PermuteTo<Tensor3D<M, O, N, H>, (Axis<0>, Axis<2>, Axis<1>)> for Tensor3D<M, N, O, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> PermuteTo<Tensor3D<N, M, O, H>, (Axis<1>, Axis<0>, Axis<2>)> for Tensor3D<M, N, O, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> PermuteTo<Tensor3D<N, O, M, H>, (Axis<1>, Axis<2>, Axis<0>)> for Tensor3D<M, N, O, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> PermuteTo<Tensor3D<O, M, N, H>, (Axis<2>, Axis<0>, Axis<1>)> for Tensor3D<M, N, O, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> PermuteTo<Tensor3D<O, N, M, H>, (Axis<2>, Axis<1>, Axis<0>)> for Tensor3D<M, N, O, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<M, N, O, P, H>, (Axis<0>, Axis<1>, Axis<2>, Axis<3>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<M, N, P, O, H>, (Axis<0>, Axis<1>, Axis<3>, Axis<2>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<M, O, N, P, H>, (Axis<0>, Axis<2>, Axis<1>, Axis<3>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<M, O, P, N, H>, (Axis<0>, Axis<2>, Axis<3>, Axis<1>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<M, P, N, O, H>, (Axis<0>, Axis<3>, Axis<1>, Axis<2>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<M, P, O, N, H>, (Axis<0>, Axis<3>, Axis<2>, Axis<1>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<N, M, O, P, H>, (Axis<1>, Axis<0>, Axis<2>, Axis<3>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<N, M, P, O, H>, (Axis<1>, Axis<0>, Axis<3>, Axis<2>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<N, O, M, P, H>, (Axis<1>, Axis<2>, Axis<0>, Axis<3>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<N, O, P, M, H>, (Axis<1>, Axis<2>, Axis<3>, Axis<0>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<N, P, M, O, H>, (Axis<1>, Axis<3>, Axis<0>, Axis<2>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<N, P, O, M, H>, (Axis<1>, Axis<3>, Axis<2>, Axis<0>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<O, M, N, P, H>, (Axis<2>, Axis<0>, Axis<1>, Axis<3>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<O, M, P, N, H>, (Axis<2>, Axis<0>, Axis<3>, Axis<1>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<O, N, M, P, H>, (Axis<2>, Axis<1>, Axis<0>, Axis<3>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<O, N, P, M, H>, (Axis<2>, Axis<1>, Axis<3>, Axis<0>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<O, P, M, N, H>, (Axis<2>, Axis<3>, Axis<0>, Axis<1>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<O, P, N, M, H>, (Axis<2>, Axis<3>, Axis<1>, Axis<0>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<P, M, N, O, H>, (Axis<3>, Axis<0>, Axis<1>, Axis<2>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<P, M, O, N, H>, (Axis<3>, Axis<0>, Axis<2>, Axis<1>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<P, N, M, O, H>, (Axis<3>, Axis<1>, Axis<0>, Axis<2>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<P, N, O, M, H>, (Axis<3>, Axis<1>, Axis<2>, Axis<0>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<P, O, M, N, H>, (Axis<3>, Axis<2>, Axis<0>, Axis<1>)> for Tensor4D<M, N, O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> PermuteTo<Tensor4D<P, O, N, M, H>, (Axis<3>, Axis<2>, Axis<1>, Axis<0>)> for Tensor4D<M, N, O, P, H>