Struct burn_ndarray::NdArrayTensor

pub struct NdArrayTensor<E, const D: usize> {
    pub array: ArcArray<E, IxDyn>,
}

Tensor primitive used by the ndarray backend.

Fields

array: ArcArray<E, IxDyn>

Dynamic array that contains the data of type E.

Implementations

impl<E, const D: usize> NdArrayTensor<E, D>

pub fn new(array: ArcArray<E, IxDyn>) -> Self

Constructs a new NdArrayTensor.

impl<E, const D: usize> NdArrayTensor<E, D>

where E: Default + Clone,

pub fn from_data(data: Data<E, D>) -> NdArrayTensor<E, D>

Create a new ndarray tensor from data.

Trait Implementations

impl<E: Clone, const D: usize> Clone for NdArrayTensor<E, D>

fn clone(&self) -> NdArrayTensor<E, D>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<E: Debug, const D: usize> Debug for NdArrayTensor<E, D>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.