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.