Enum Storage 

pub enum Storage {
    Cpu(CpuStorage),
}

Variants

Cpu(CpuStorage)

Trait Implementations

impl ActivationOps for Storage

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 Add<&Storage> for &Storage

type Output = Storage

The resulting type after applying the + operator.

fn add(self, rhs: &Storage) -> Self::Output

Performs the + operation.

impl Add<f32> for &Storage

type Output = Storage

The resulting type after applying the + operator.

fn add(self, rhs: f32) -> Self::Output

Performs the + operation.

impl AddAssign<&Storage> for Storage

fn add_assign(&mut self, rhs: &Storage)

Performs the += operation.

impl AddAssign<f32> for Storage

fn add_assign(&mut self, rhs: f32)

Performs the += operation.

impl ArithmeticOps for Storage

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

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

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

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

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

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

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

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

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

fn add_f32_assign(&mut self, other: f32)

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

fn sub_f32_assign(&mut self, other: f32)

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

fn mul_f32_assign(&mut self, other: f32)

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

fn div_f32_assign(&mut self, other: f32)

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

fn pow_f32_assign(&mut self, other: f32)

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 abs_assign(&mut self)

impl BlasOps for Storage

fn matmul(&self, other: &Self, trans_a: bool, trans_b: bool) -> Self

impl Clone for Storage

fn clone(&self) -> Storage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Storage

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

Formats the value using the given formatter.

impl DeviceStorage for Storage

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

where Self: Sized,

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 Storage

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 Display for Storage

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

Formats the value using the given formatter.

impl Display for Storage

fn print(&self)

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

fn print_data(&self)

impl Div<&Storage> for &Storage

type Output = Storage

The resulting type after applying the / operator.

fn div(self, rhs: &Storage) -> Self::Output

Performs the / operation.

impl Div<f32> for &Storage

type Output = Storage

The resulting type after applying the / operator.

fn div(self, rhs: f32) -> Self::Output

Performs the / operation.

impl DivAssign<&Storage> for Storage

fn div_assign(&mut self, rhs: &Storage)

Performs the /= operation.

impl DivAssign<f32> for Storage

fn div_assign(&mut self, rhs: f32)

Performs the /= operation.

impl Mul<&Storage> for &Storage

type Output = Storage

The resulting type after applying the * operator.

fn mul(self, rhs: &Storage) -> Self::Output

Performs the * operation.

impl Mul<f32> for &Storage

type Output = Storage

The resulting type after applying the * operator.

fn mul(self, rhs: f32) -> Self::Output

Performs the * operation.

impl MulAssign<&Storage> for Storage

fn mul_assign(&mut self, rhs: &Storage)

Performs the *= operation.

impl MulAssign<f32> for Storage

fn mul_assign(&mut self, rhs: f32)

Performs the *= operation.

impl ReductionOps for Storage

fn sum(&self) -> Self

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

fn product(&self) -> Self

fn mean(&self) -> Self

impl Sub<&Storage> for &Storage

type Output = Storage

The resulting type after applying the - operator.

fn sub(self, rhs: &Storage) -> Self::Output

Performs the - operation.

impl Sub<f32> for &Storage

type Output = Storage

The resulting type after applying the - operator.

fn sub(self, rhs: f32) -> Self::Output

Performs the - operation.

impl SubAssign<&Storage> for Storage

fn sub_assign(&mut self, rhs: &Storage)

Performs the -= operation.

impl SubAssign<f32> for Storage

fn sub_assign(&mut self, rhs: f32)

Performs the -= operation.

impl TransformOps for Storage

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

where F: Fn(f32) -> f32,

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

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 elementwise_op_assign<F>(&mut self, other: &Self, op: F)

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

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

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

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

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

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

fn transpose(&self) -> Self

fn flatten(&mut self)

fn squeeze(&mut self)

fn unsqueeze(&mut self, dim: usize)

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

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

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

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

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

where Self: Sized,