Struct burn_core::tensor::Data

pub struct Data<E, const D: usize> {
    pub value: Vec<E, Global>,
    pub shape: Shape<D>,
}

Fields

value: Vec<E, Global>

shape: Shape<D>

Implementations

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

pub fn new(value: Vec<E, Global>, shape: Shape<D>) -> Data<E, D>

Constructs a new Data.

impl<const D: usize, E> Data<E, D>where E: Element,

pub fn convert<EOther>(self) -> Data<EOther, D>where EOther: Element,

impl<const D: usize> Data<bool, D>

pub fn convert<E>(self) -> Data<E, D>where E: Element,

impl<E, const D: usize> Data<E, D>where E: Element,

pub fn random<R>( shape: Shape<D>, distribution: Distribution<E>, rng: &mut R ) -> Data<E, D>where R: RngCore,

impl<E, const D: usize> Data<E, D>where E: Debug + Element,

pub fn zeros<S>(shape: S) -> Data<E, D>where S: Into<Shape<D>>,

pub fn zeros_(shape: Shape<D>, _kind: E) -> Data<E, D>

impl<E, const D: usize> Data<E, D>where E: Debug + Element,

pub fn ones(shape: Shape<D>) -> Data<E, D>

pub fn ones_(shape: Shape<D>, _kind: E) -> Data<E, D>

impl<E, const D: usize> Data<E, D>where E: Debug + Copy,

pub fn serialize(&self) -> DataSerialize<E>

impl<E, const D: usize> Data<E, D>where E: Into<f64> + Clone + Debug + PartialEq<E>,

pub fn assert_approx_eq(&self, other: &Data<E, D>, precision: usize)

pub fn assert_in_range(&self, min: E, max: E)

impl<const D: usize> Data<usize, D>

pub fn from_usize<O>(self) -> Data<O, D>where O: FromPrimitive,

Trait Implementations

impl<E, const D: usize> Clone for Data<E, D>where E: Clone,

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

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<E, const D: usize> Debug for Data<E, D>where E: Debug,

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

Formats the value using the given formatter.

impl<E, const D: usize> Display for Data<E, D>where E: Debug,

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

Formats the value using the given formatter.

impl<E> From<&[E]> for Data<E, 1>where E: Debug + Copy,

fn from(elems: &[E]) -> Data<E, 1>

Converts to this type from the input type.

impl<E, const D: usize> From<&DataSerialize<E>> for Data<E, D>where E: Clone,

fn from(data: &DataSerialize<E>) -> Data<E, D>

Converts to this type from the input type.

impl<E, const A: usize, const B: usize, const C: usize, const D: usize> From<[[[[E; D]; C]; B]; A]> for Data<E, 4>where E: Debug + Copy,

fn from(elems: [[[[E; D]; C]; B]; A]) -> Data<E, 4>

Converts to this type from the input type.

impl<E, const A: usize, const B: usize, const C: usize> From<[[[E; C]; B]; A]> for Data<E, 3>where E: Debug + Copy,

fn from(elems: [[[E; C]; B]; A]) -> Data<E, 3>

Converts to this type from the input type.

impl<E, const A: usize, const B: usize> From<[[E; B]; A]> for Data<E, 2>where E: Debug + Copy,

fn from(elems: [[E; B]; A]) -> Data<E, 2>

Converts to this type from the input type.

impl<E, const A: usize> From<[E; A]> for Data<E, 1>where E: Debug + Copy,

fn from(elems: [E; A]) -> Data<E, 1>

Converts to this type from the input type.

impl<E, const D: usize> From<DataSerialize<E>> for Data<E, D>

fn from(data: DataSerialize<E>) -> Data<E, D>

Converts to this type from the input type.

impl<E, const D: usize> PartialEq<Data<E, D>> for Data<E, D>where E: PartialEq<E>,

fn eq(&self, other: &Data<E, D>) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<E, const D: usize> Eq for Data<E, D>where E: Eq,

impl<E, const D: usize> StructuralEq for Data<E, D>

impl<E, const D: usize> StructuralPartialEq for Data<E, D>