code: def create_inputs(rng, axis, device_id):
      x = rng.randn(2, 3, 4).astype(np.float32) * 2
      x += device_id
      shape_stat = [1 for _ in range(x.ndim)]
      shape_stat[axis] = x.shape[axis]
      beta = rng.randn(*shape_stat).astype(np.float32)
      gamma = rng.randn(*shape_stat).astype(np.float32)
      rmean = np.zeros(shape_stat, dtype=np.float32)
      rvar = np.zeros(shape_stat, dtype=np.float32)
      return x, beta, gamma, rmean, rvar
  
file_path: nnabla/python/test/communicator/test_sync_batch_normalization.py
imports: import pytest
  import numpy
  import nnabla
  import nnabla.functions
  from nbla_test_utils import list_context
  from nnabla.testing import assert_allclose
  from six.moves import range
label_desc: Use image node_id 1 for calling a global function with example usage: create_inputs(rng, axis, device_id) and returns: x,  beta,  gamma,  rmean,  rvar