Trait tensor_rs::tensor_trait::reduction::ReduceTensor

source · [−]
pub trait ReduceTensor where
    Self: Sized, {

Show 17 methods    fn argmax(&self, dim: Option<&[usize]>, keep_dim: bool) -> Self;

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

    fn dist();

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

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

    fn median();

    fn mode();

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

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

    fn std_mean();

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

    fn unique();

    fn unique_consecutive();

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

    fn var_mean();

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

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

}

Required methods

log(sum(exp(x))), dim is the dimension along which sum is applied. if keep_dim, the dimension along which sum is applied will be kept and be 1.

Implementors