code: def batch_norm_1d(num_channels):
      if dist.get_world_size() > 1:
          return nn.SyncBatchNorm.convert_sync_batchnorm(
              nn.BatchNorm1D(num_channels)
          )
      else:
          return nn.BatchNorm1D(num_channels)
  
imports: import numpy
  import math
  import paddle
  import paddle.nn
  import paddle.nn.functional
  import paddle.distributed
  import pgl
  import pgl.nn
  from pgl.nn import functional
  from pgl.utils.logger import log
  import models.mol_encoder
label_desc: Use image node_id 1 for calling a global function with example usage: batch_norm_1d(num_channels) and returns: nn, nn