Trait ToNdArray

pub trait ToNdArray<B: Backend, const D: usize, T> {
    // Required methods
    fn to_ndarray(&self) -> Array<T, Dim<[usize; D]>>;
    fn into_ndarray(self) -> Array<T, Dim<[usize; D]>>;
}

Trait for converting burn tensor to ndarray (generic over Float/Int and dimensionality)

Required Methods

fn to_ndarray(&self) -> Array<T, Dim<[usize; D]>>

fn into_ndarray(self) -> Array<T, Dim<[usize; D]>>

Implementations on Foreign Types

impl<B: Backend> ToNdArray<B, 1, f32> for Tensor<B, 1, Float>

Trait implementation for converting 1D Float burn tensor to ndarray

fn to_ndarray(&self) -> Array1<f32>

fn into_ndarray(self) -> Array1<f32>

impl<B: Backend> ToNdArray<B, 1, u32> for Tensor<B, 1, Int>

Trait implementation for converting 1D Int burn tensor to ndarray

fn to_ndarray(&self) -> Array1<u32>

fn into_ndarray(self) -> Array1<u32>

impl<B: Backend> ToNdArray<B, 2, f32> for Tensor<B, 2, Float>

Trait implementation for converting 2D Float burn tensor to ndarray

fn to_ndarray(&self) -> Array2<f32>

fn into_ndarray(self) -> Array2<f32>

impl<B: Backend> ToNdArray<B, 2, u32> for Tensor<B, 2, Int>

Trait implementation for converting 2D Int burn tensor to ndarray

fn to_ndarray(&self) -> Array2<u32>

fn into_ndarray(self) -> Array2<u32>

impl<B: Backend> ToNdArray<B, 3, f32> for Tensor<B, 3, Float>

Trait implementation for converting 3D Float burn tensor to ndarray

fn to_ndarray(&self) -> Array3<f32>

fn into_ndarray(self) -> Array3<f32>

impl<B: Backend> ToNdArray<B, 3, u32> for Tensor<B, 3, Int>

Trait implementation for converting 3D Int burn tensor to ndarray

fn to_ndarray(&self) -> Array3<u32>

fn into_ndarray(self) -> Array3<u32>

Implementors