stonnx_api::operators::batchnormalization

Function batchnormalization

Source 
pub fn batchnormalization(
    inputs: &[&TensorType],
    node: &NodeProto,
    opset_version: i64,
    _output_len: usize,
) -> BoxResult<OperatorResult>
Expand description

Carries out batch normalization as described in the paper https://arxiv.org/abs/1502.03167.

Depending on the mode it is being run, There are five required inputs ‘X’, ‘scale’, ‘B’, ‘input_mean’ and ‘input_var’.

Python reference

ONNX Documentation