code: def _batchnorm_test_mode(
      x: np.ndarray,
      s: np.ndarray,
      bias: np.ndarray,
      mean: np.ndarray,
      var: np.ndarray,
      epsilon: float = 1e-5,
  ) -> np.ndarray:
      dims_x = len(x.shape)
      dim_ones = (1,) * (dims_x - 2)
      s = s.reshape(-1, *dim_ones)
      bias = bias.reshape(-1, *dim_ones)
      mean = mean.reshape(-1, *dim_ones)
      var = var.reshape(-1, *dim_ones)
      y = s * (x - mean) / np.sqrt(var + epsilon) + bias
      return y.astype(x.dtype)
  
file_path: onnx/onnx/reference/ops/op_batch_normalization.py
imports: import numpy
  from onnx.reference.op_run import OpRun
label_desc: Use image node_id 1 for calling a global function with example usage: _batchnorm_test_mode(x, s, bias, mean, var, epsilon) and returns: y