Trait ReductionOps 

pub trait ReductionOps {
    // Required methods
    fn sum_all(&self) -> Result<Tensor>;
    fn sum_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>;
    fn mean_all(&self) -> Result<Tensor>;
    fn mean_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>;
    fn max_all(&self) -> Result<Tensor>;
    fn max_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>;
    fn min_all(&self) -> Result<Tensor>;
    fn min_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>;
    fn prod_all(&self) -> Result<Tensor>;
    fn std_all(&self) -> Result<Tensor>;
    fn var_all(&self) -> Result<Tensor>;
    fn norm(&self) -> Result<Tensor>;
    fn norm_p(&self, p: f32) -> Result<Tensor>;
}

Trait for reduction operations on tensors.

Required Methods

fn sum_all(&self) -> Result<Tensor>

Sum all elements in the tensor.

fn sum_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>

Sum along specified dimensions.

fn mean_all(&self) -> Result<Tensor>

Mean of all elements in the tensor.

fn mean_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>

Mean along specified dimensions.

fn max_all(&self) -> Result<Tensor>

Maximum value in the tensor.

fn max_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>

Maximum along specified dimensions.

fn min_all(&self) -> Result<Tensor>

Minimum value in the tensor.

fn min_dims(&self, dims: &[usize], keep_dim: bool) -> Result<Tensor>

Minimum along specified dimensions.

fn prod_all(&self) -> Result<Tensor>

Product of all elements.

fn std_all(&self) -> Result<Tensor>

Standard deviation of all elements.

fn var_all(&self) -> Result<Tensor>

Variance of all elements.

fn norm(&self) -> Result<Tensor>

L2 norm (Euclidean norm) of the tensor.

fn norm_p(&self, p: f32) -> Result<Tensor>

Lp norm of the tensor.

Implementors

impl ReductionOps for Tensor