Trait ToBurn

pub trait ToBurn<B: Backend, const D: usize, T: TensorKind<B>> {
    // Required methods
    fn to_burn(&self, device: &B::Device) -> Tensor<B, D, T>;
    fn into_burn(self, device: &B::Device) -> Tensor<B, D, T>;
}

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

Required Methods

fn to_burn(&self, device: &B::Device) -> Tensor<B, D, T>

fn into_burn(self, device: &B::Device) -> Tensor<B, D, T>

Implementations on Foreign Types

impl<B: Backend> ToBurn<B, 1, Float> for Array1<f32>

Trait implementation for 1D Float ndarray

fn to_burn(&self, device: &B::Device) -> Tensor<B, 1, Float>

fn into_burn(self, device: &B::Device) -> Tensor<B, 1, Float>

impl<B: Backend> ToBurn<B, 1, Int> for Array1<u32>

Trait implementation for 1D Int ndarray

fn to_burn(&self, device: &B::Device) -> Tensor<B, 1, Int>

fn into_burn(self, device: &B::Device) -> Tensor<B, 1, Int>

impl<B: Backend> ToBurn<B, 2, Float> for DMatrix<f32>

Implement ToBurn for converting nalgebra::DMatrix<f32> to a burn tensor (Float type)

fn to_burn(&self, device: &B::Device) -> Tensor<B, 2, Float>

fn into_burn(self, device: &B::Device) -> Tensor<B, 2, Float>

impl<B: Backend> ToBurn<B, 2, Float> for Array2<f32>

Implementation of the trait for 2D Float ndarray

fn to_burn(&self, device: &B::Device) -> Tensor<B, 2, Float>

fn into_burn(self, device: &B::Device) -> Tensor<B, 2, Float>

impl<B: Backend> ToBurn<B, 2, Int> for DMatrix<u32>

Implement ToBurn for converting nalgebra::DMatrix<u32> to a burn tensor (Int type)

fn to_burn(&self, device: &B::Device) -> Tensor<B, 2, Int>

fn into_burn(self, device: &B::Device) -> Tensor<B, 2, Int>

impl<B: Backend> ToBurn<B, 2, Int> for Array2<u32>

Trait implementation for 2D Int ndarray

fn to_burn(&self, device: &B::Device) -> Tensor<B, 2, Int>

fn into_burn(self, device: &B::Device) -> Tensor<B, 2, Int>

impl<B: Backend> ToBurn<B, 3, Float> for Array3<f32>

fn to_burn(&self, device: &B::Device) -> Tensor<B, 3, Float>

fn into_burn(self, device: &B::Device) -> Tensor<B, 3, Float>

impl<B: Backend> ToBurn<B, 3, Int> for Array3<u32>

Trait implementation for 3D Int ndarray

fn to_burn(&self, device: &B::Device) -> Tensor<B, 3, Int>

fn into_burn(self, device: &B::Device) -> Tensor<B, 3, Int>

Implementors