Trait dfdx::tensor::TensorToArray

pub trait TensorToArray<S: Shape, E>: Storage<E> {
    type Array: Debug + PartialEq;

    // Required method
    fn tensor_to_array<T>(&self, tensor: &Tensor<S, E, Self, T>) -> Self::Array;
}

Required Associated Types

type Array: Debug + PartialEq

Required Methods

fn tensor_to_array<T>(&self, tensor: &Tensor<S, E, Self, T>) -> Self::Array

Implementors

impl<E: Unit> TensorToArray<(), E> for Cpu

type Array = E

impl<E: Unit, const M: usize> TensorToArray<(Const<M>,), E> for Cpu

type Array = [E; M]

impl<E: Unit, const M: usize, const N: usize> TensorToArray<(Const<M>, Const<N>), E> for Cpu

type Array = [[E; N]; M]

impl<E: Unit, const M: usize, const N: usize, const O: usize> TensorToArray<(Const<M>, Const<N>, Const<O>), E> for Cpu

type Array = [[[E; O]; N]; M]

impl<E: Unit, const M: usize, const N: usize, const O: usize, const P: usize> TensorToArray<(Const<M>, Const<N>, Const<O>, Const<P>), E> for Cpu

type Array = [[[[E; P]; O]; N]; M]

impl<S: Shape, E: Unit> TensorToArray<S, E> for Cudawhere Cpu: TensorToArray<S, E> + Storage<E>,

type Array = <Cpu as TensorToArray<S, E>>::Array