Struct dfdx::tensor::Cpu

pub struct Cpu { /* private fields */ }

A device that stores data on the heap.

The Default impl seeds the underlying rng with seed of 0.

Use Cpu::seed_from_u64 to control what seed is used.

Implementations

impl Cpu

pub fn seed_from_u64(seed: u64) -> Self

Constructs rng with the given seed.

Trait Implementations

impl<E: Float + Dtype> AdamKernel<E> for Cpu

fn update( &self, t: i32, cfg: &AdamConfig<E>, param: &mut Self::Vec<E>, moment1: &mut Self::Vec<E>, moment2: &mut Self::Vec<E>, grad: &Self::Vec<E> ) -> Result<(), Self::Err>

impl Clone for Cpu

fn clone(&self) -> Cpu

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<E: Unit> CopySlice<E> for Cpu

fn copy_from<S: Shape, T>(dst: &mut Tensor<S, E, Self, T>, src: &[E])

fn copy_into<S: Shape, T>(src: &Tensor<S, E, Self, T>, dst: &mut [E])

impl Debug for Cpu

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

Formats the value using the given formatter.

impl Default for Cpu

fn default() -> Self

Returns the “default value” for a type.

impl DeviceStorage for Cpu

type Vec<E: Unit> = Vec<E, Global>

Generic storage type

fn try_alloc_grad<E: Unit>( &self, other: &Self::Vec<E> ) -> Result<Self::Vec<E>, Self::Err>

Allocates a gradient for the given nd array

fn random_u64(&self) -> u64

Generates a random u64 number

fn tensor_to_vec<S: Shape, E: Unit, T>( &self, tensor: &Tensor<S, E, Self, T> ) -> Vec<E>

impl HasErr for Cpu

type Err = CpuError

impl<E: Unit> OnesTensor<E> for Cpu

fn try_ones_like<S: HasShape>( &self, src: &S ) -> Result<Tensor<S::Shape, E, Self>, Self::Err>

Fallible version of OnesTensor::ones_like

fn ones<S: ConstShape>(&self) -> Tensor<S, E, Self>

Creates a tensor filled with ones.

fn try_ones<S: ConstShape>(&self) -> Result<Tensor<S, E, Self>, Self::Err>

Fallible version of OnesTensor::ones

fn ones_like<S: HasShape>(&self, src: &S) -> Tensor<S::Shape, E, Self>

Build the tensor with a shape given by something else.

impl<E: Float + Dtype> RMSpropKernel<E> for Cpu

fn update( &self, cfg: &RMSpropConfig<E>, param: &mut Self::Vec<E>, momentum: &mut Self::Vec<E>, square_avg: &mut Self::Vec<E>, grad_avg: &mut Self::Vec<E>, grad: &Self::Vec<E> ) -> Result<(), Self::Err>

impl<E: Unit> SampleTensor<E> for Cpu

fn try_sample_like<S: HasShape, D: Distribution<E>>( &self, src: &S, distr: D ) -> Result<Tensor<S::Shape, E, Self>, Self::Err>

Fallibly samples a tensor with a given shape from a given distribution.

fn try_fill_with_distr<D: Distribution<E>>( &self, storage: &mut Self::Vec<E>, distr: D ) -> Result<(), Self::Err>

Fills tensor storage with data from a given distribution

fn sample<S: ConstShape, D: Distribution<E>>( &self, distr: D ) -> Tensor<S, E, Self>

Samples a const tensor from a given distribution.

fn try_sample<S: ConstShape, D: Distribution<E>>( &self, distr: D ) -> Result<Tensor<S, E, Self>, Self::Err>

Fallibly samples a const tensor from a given distribution.

fn sample_like<S: HasShape, D: Distribution<E>>( &self, src: &S, distr: D ) -> Tensor<S::Shape, E, Self>

Samples a tensor with a given shape from a given distribution.

impl<E: Dtype> SgdKernel<E> for Cpu

fn update( &self, cfg: &SgdConfig<E>, param: &mut Self::Vec<E>, velocity: &mut Self::Vec<E>, grad: &Self::Vec<E> ) -> Result<(), Self::Err>

impl<E: Unit> TensorFromVec<E> for Cpu

fn try_tensor_from_vec<S: Shape>( &self, src: Vec<E>, shape: S ) -> Result<Tensor<S, E, Self>, Self::Err>

fn tensor_from_vec<S: Shape>(&self, src: Vec<E>, shape: S) -> Tensor<S, E, Self>

impl<E: Unit> ZerosTensor<E> for Cpu

fn try_zeros_like<S: HasShape>( &self, src: &S ) -> Result<Tensor<S::Shape, E, Self>, Self::Err>

Fallible version of ZerosTensor::zeros_like

fn zeros<S: ConstShape>(&self) -> Tensor<S, E, Self>

Creates a tensor filled with zeros.

fn try_zeros<S: ConstShape>(&self) -> Result<Tensor<S, E, Self>, Self::Err>

Fallible version of ZerosTensor::zeros

fn zeros_like<S: HasShape>(&self, src: &S) -> Tensor<S::Shape, E, Self>

Build the tensor with a shape given by something else.

impl Device<f32> for Cpu

impl Device<f64> for Cpu