Struct CpuStorage 

pub struct CpuStorage { /* private fields */ }

Trait Implementations

impl ActivationOps for CpuStorage

fn binary_step(&self) -> Self

fn sigmoid(&self) -> Self

fn tanh(&self) -> Self

fn relu(&self) -> Self

fn leaky_relu(&self) -> Self

fn parametric_relu(&self, a: f32) -> Self

fn elu(&self, alpha: f32) -> Self

fn softmax(&self, dim: usize) -> Self

fn swish(&self) -> Self

impl ArithmeticOps for CpuStorage

fn add_tensor(&self, other: &Self) -> Self

fn sub_tensor(&self, other: &Self) -> Self

fn mul_tensor(&self, other: &Self) -> Self

fn div_tensor(&self, other: &Self) -> Self

fn add_f32(&self, other: f32) -> Self

fn sub_f32(&self, other: f32) -> Self

fn mul_f32(&self, other: f32) -> Self

fn div_f32(&self, other: f32) -> Self

fn pow_f32(&self, other: f32) -> Self

fn greater_than(&self, other: &Self, make_binary: bool) -> Self

fn greater_than_f32(&self, other: f32, make_binary: bool) -> Self

fn less_than(&self, other: &Self, make_binary: bool) -> Self

fn less_than_f32(&self, other: f32, make_binary: bool) -> Self

fn sign(&self) -> Self

fn abs(&self) -> Self

fn add_tensor_assign(&mut self, other: &Self)

fn sub_tensor_assign(&mut self, other: &Self)

fn mul_tensor_assign(&mut self, other: &Self)

fn div_tensor_assign(&mut self, other: &Self)

fn add_f32_assign(&mut self, other: f32)

fn sub_f32_assign(&mut self, other: f32)

fn mul_f32_assign(&mut self, other: f32)

fn div_f32_assign(&mut self, other: f32)

fn pow_f32_assign(&mut self, other: f32)

fn abs_assign(&mut self)

impl BlasOps for CpuStorage

fn matmul( &self, other: &Self, transpose_self: bool, transpose_other: bool, ) -> Self

impl Clone for CpuStorage

fn clone(&self) -> CpuStorage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CpuStorage

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

Formats the value using the given formatter.

impl DeviceStorage for CpuStorage

fn view(&self, new_shape: Vec<usize>) -> Self

fn data(&self) -> Arc<RwLock<Vec<f32>>>

fn data_mut(&self) -> RwLockWriteGuard<'_, Vec<f32>>

fn set_data(&mut self, data: Vec<f32>)

fn shape(&self) -> &Vec<usize>

fn set_shape(&mut self, shape: Vec<usize>)

fn stride(&self) -> &Vec<usize>

fn set_stride(&mut self, stride: Vec<usize>)

fn offset(&self) -> usize

fn get(&self, indices: &[usize]) -> f32

fn set(&mut self, indices: &[usize], value: f32)

fn make_contiguous(&self) -> (Vec<f32>, i32)

fn is_contiguous(&self) -> bool

impl DeviceStorageCreation for CpuStorage

fn zeros( shape: Vec<usize>, _device: Option<Device>, _requires_grad: Option<bool>, ) -> Self

fn ones( shape: Vec<usize>, _device: Option<Device>, _requires_grad: Option<bool>, ) -> Self

fn from_ndarray<S, D, T>( data: &ArrayBase<S, D>, _device: Option<Device>, _requires_grad: Option<bool>, ) -> Self

where S: Data<Elem = T>, T: AsPrimitive<f32>, D: Dimension,

fn uniform( l_bound: f32, r_bound: f32, shape: Vec<usize>, _device: Option<Device>, _requires_grad: Option<bool>, ) -> Self

impl DeviceStorageStatic for CpuStorage

fn new(data: Vec<f32>, shape: Vec<usize>) -> Self

fn new_with_stride( data: Vec<f32>, shape: Vec<usize>, stride: Vec<usize>, ) -> Self

fn create( data: Arc<RwLock<Vec<f32>>>, shape: Vec<usize>, stride: Vec<usize>, ) -> Self

fn compute_strides(shape: &Vec<usize>) -> Vec<usize>

impl Display for CpuStorage

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

Formats the value using the given formatter.

impl Display for CpuStorage

fn print(&self)

fn print_data_recursive<'a>( data: &'a [f32], shape: &'a [usize], stride: &'a [usize], ) -> String

fn print_data(&self)

impl ReductionOps for CpuStorage

fn sum(&self) -> Self

fn sum_axis(&self, axis: usize) -> Self

fn product(&self) -> Self

fn mean(&self) -> Self

impl TransformOps for CpuStorage

fn compute_broadcast_shape(&self, target_shape: &[usize]) -> Vec<usize>

Compute broadcast shape between two shapes

fn compute_broadcast_strides(&self, broadcast_shape: &[usize]) -> Vec<usize>

Compute broadcast strides

fn pad_shape(&self, target_rank: usize) -> Vec<usize>

Pad shape with ones on the left

fn apply_assign<F>(&mut self, op: F)

where F: Fn(f32) -> f32,

fn elementwise_op_assign<F>(&mut self, other: &Self, op: F)

where F: Fn(f32, f32) -> f32,

fn reshape(&mut self, new_shape: Vec<usize>)

fn scalar_op_assign<F>(&mut self, scalar: f32, op: F)

where F: Fn(f32, f32) -> f32,

fn permute(&mut self, dims: &[usize])

fn flatten(&mut self)

fn squeeze(&mut self)

fn unsqueeze(&mut self, dim: usize)

fn apply<F>(&self, op: F) -> Self

where F: Fn(f32) -> f32,

fn elementwise_op<F>(&self, other: &Self, op: F) -> Self

where F: Fn(f32, f32) -> f32,

fn scalar_op<F>(&self, scalar: f32, op: F) -> Self

where F: Fn(f32, f32) -> f32,

fn sum_dim(&self, dims: &[bool]) -> Self

fn transpose(&self) -> Self

fn broadcast(&self, new_shape: &[usize]) -> Self

fn broadcast_tensors(a: &Self, b: &Self) -> (Self, Self)