product_by_key_nan

Function product_by_key_nan 

Source
pub fn product_by_key_nan<KeyType, ValueType>(
    keys: &Array<KeyType>,
    vals: &Array<ValueType>,
    dim: i32,
    replace_value: f64,
) -> (Array<KeyType>, Array<ValueType::ProductOutType>)
where KeyType: ReduceByKeyInput, ValueType: HasAfEnum, ValueType::ProductOutType: HasAfEnum,
Expand description

Compute product of all values with similar keys along a given dimension

This version of sum by key can replaced all NaN values in the input with a user provided value before performing the reduction operation.

§Parameters

  • keys - key Array
  • vals - value Array
  • dim - Dimension along which the input Array is reduced

§Return Values

Tuple of Arrays, with output keys and values after reduction