code: def _batchnorm_test_mode(x, s, bias, mean, var, epsilon=1e-5):  # type: ignore
      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)
      return s * (x - mean) / np.sqrt(var + epsilon) + bias
  
file_path: onnx/onnx/backend/test/case/node/batchnorm.py
imports: import numpy
  import onnx
  from onnx.backend.test.case.base import Base
  from onnx.backend.test.case.node import expect
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: unknown