[−][src]Trait auto_diff::tensor::reduction::ReduceTensor

pub trait ReduceTensor {
    type TensorType;
    fn argmax(&self, dim: Option<&[usize]>, keep_dim: bool) -> Self::TensorType;

    fn argmin(&self, dim: Option<&[usize]>, keep_dim: bool) -> Self::TensorType;

    fn dist();

    fn logsumexp(
        &self, 
        dim: Option<&[usize]>, 
        keep_dim: bool
    ) -> Self::TensorType;

    fn mean(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;

    fn median();

    fn mode();

    fn norm();

    fn prod(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;

    fn std(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;

    fn std_mean();

    fn sum(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;

    fn unique();

    fn unique_consecutive();

    fn var(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;

    fn var_mean();

    fn max(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;

    fn min(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType;
}

Associated Types

type TensorType

Loading content...

Required methods

fn argmax(&self, dim: Option<&[usize]>, keep_dim: bool) -> Self::TensorType

fn argmin(&self, dim: Option<&[usize]>, keep_dim: bool) -> Self::TensorType

fn dist()

fn logsumexp(&self, dim: Option<&[usize]>, keep_dim: bool) -> Self::TensorType

fn mean(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

fn median()

fn mode()

fn norm()

fn prod(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

fn std(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

fn std_mean()

fn sum(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

fn unique()

fn unique_consecutive()

fn var(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

fn var_mean()

fn max(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

fn min(&self, dim: Option<&[usize]>, keepdim: bool) -> Self::TensorType

Loading content...

Implementors

impl<T> ReduceTensor for GenTensor<T> where
    T: Float, [src]

type TensorType = GenTensor<T>

fn mean(&self, dim: Option<&[usize]>, keep_dim: bool) -> GenTensor<T>[src]

Returns the mean value of the tensor along dim row.

fn sum(&self, dim: Option<&[usize]>, keep_dim: bool) -> GenTensor<T>[src]

Returns the sum of all elements.

let m1 = GenTensor::<f64>::new_raw(&vec![1.,2.,3.,4.,], &vec![2,2]);
assert_eq!(m1.sum(None, false).get_scale(), 10.);
Loading content...