code: def batch_normalization_warp(y):
      _, output_size = y.get_shape()
      output_size = int(output_size)
      out_F = int(output_size / IN_JOINTS)
      y = Reshape([-1, IN_JOINTS, out_F])(y)
      y = BatchNorm(act="lrelu", epsilon=1e-3)(y)
      y = Reshape([-1, output_size])(y)
      return y
  
imports: import numpy
  import tensorflow
  from tensorlayer.layers import Layer, Dropout, Dense, Input, BatchNorm, Reshape, Elementwise
  from tensorlayer.models import Model
  from tensorlayer import logging
  from .common import mask_weight, neighbour_matrix
label_desc: Use image node_id 1 for calling a global function with example usage: batch_normalization_warp(y) and returns: y