Struct burn_tensor::Data

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

Fields

value: Vec<P>

shape: Shape<D>

Implementations

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

pub fn new(value: Vec<P>, shape: Shape<D>) -> Self

Constructs a new Data.

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

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

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

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

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

pub fn random(
    shape: Shape<D>,
    distribution: Distribution<P>,
    rng: &mut StdRng
) -> Self

impl<P, const D: usize> Data<P, D>where
    P: Zeros + Default + Debug,

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

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

impl<P, const D: usize> Data<P, D>where
    P: Ones + Default + Debug,

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

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

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

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

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

pub fn assert_approx_eq(&self, other: &Self, precision: usize)

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

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

Trait Implementations

impl<P: Clone, const D: usize> Clone for Data<P, D>

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

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<P: Debug, const D: usize> Debug for Data<P, D>

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

Formats the value using the given formatter.

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

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

Formats the value using the given formatter.

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

fn from(data: &DataSerialize<P>) -> Self

Converts to this type from the input type.

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

fn from(elems: [[[P; C]; B]; A]) -> Self

Converts to this type from the input type.

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

fn from(elems: [[P; B]; A]) -> Self

Converts to this type from the input type.

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

fn from(elems: [P; A]) -> Self

Converts to this type from the input type.

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

fn from(data: DataSerialize<P>) -> Self

Converts to this type from the input type.

impl<P: PartialEq, const D: usize> PartialEq<Data<P, D>> for Data<P, D>

fn eq(&self, other: &Data<P, 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<P: Eq, const D: usize> Eq for Data<P, D>

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

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