meanvar

Function meanvar 

Source
pub fn meanvar<T, W>(
    input: &Array<T>,
    weights: &Array<W>,
    bias: VarianceBias,
    dim: i64,
) -> (Array<T::MeanOutType>, Array<T::MeanOutType>)
Expand description

Calculate mean and variance in single API call

§Parameters

  • input is the input Array
  • weights Array has the weights to be used during the stat computation
  • bias is type of bias used for variance calculation
  • dim is dimension along which the current stat has to be computed

§Return Values

A tuple of Arrays, whose size is equal to input except along the dimension which the stat operation is performed. Array size along dim will be reduced to one.

  • First Array contains mean values
  • Second Array contains variance values