Trait dfdx::tensor::TensorCreator

pub trait TensorCreator: Sized + HasDevice {
    fn new_boxed(data: Box<Self::Array>) -> Self;

    fn new(data: Self::Array) -> Self { ... }
    fn zeros() -> Self { ... }
    fn ones() -> Self
    where
        Self::Dtype: One,
    { ... }
    fn rand<R: Rng>(rng: &mut R) -> Self
    where
        Standard: Distribution<Self::Dtype>,
    { ... }
    fn randn<R: Rng>(rng: &mut R) -> Self
    where
        StandardNormal: Distribution<Self::Dtype>,
    { ... }
}

Something that can be created - currently only implemented for tensors with no tapes.

Required Methods

fn new_boxed(data: Box<Self::Array>) -> Self

Create a new tensor with a Box<Self::Array>.

Provided Methods

fn new(data: Self::Array) -> Self

Create a new tensor with Self::Array on the stack. This just boxes Self::Array and calls TensorCreator::new_boxed.

fn zeros() -> Self

Creates a tensor filled with all 0s.

fn ones() -> Self where
    Self::Dtype: One, 

Creates a tensor filled with all 1s.

fn rand<R: Rng>(rng: &mut R) -> Self where
    Standard: Distribution<Self::Dtype>, 

Creates a tensor filled with values sampled from Standard distribution.

fn randn<R: Rng>(rng: &mut R) -> Self where
    StandardNormal: Distribution<Self::Dtype>, 

Creates a tensor filled with values sampled from StandardNormal distribution.

Implementors

impl TensorCreator for Tensor0D<NoTape>

impl<const M: usize> TensorCreator for Tensor1D<M, NoTape>

impl<const M: usize, const N: usize> TensorCreator for Tensor2D<M, N, NoTape>

impl<const M: usize, const N: usize, const O: usize> TensorCreator for Tensor3D<M, N, O, NoTape>

impl<const M: usize, const N: usize, const O: usize, const P: usize> TensorCreator for Tensor4D<M, N, O, P, NoTape>