'Sets shape based parameters of this layer given an input tuple or int
or input layer.
Arguments:
in_obj (int, tuple, Layer or Tensor): object that provides shape
information for layer
Returns:
(tuple): shape of output data'

def configure(self, in_obj):

    super(BatchNorm, self).configure(in_obj)
    self.out_shape = self.in_shape
    (self.nin, self.nsteps) = interpret_in_shape(self.in_shape)
    self.nfm = (self.in_shape[0] if isinstance(self.in_shape, tuple) else self.nin)
    self.nglayer = self.be.batchnorm_layer(self.in_shape)
    return self
