message: Update to use tf.shape to get the shape of the tensor, from review comment
diff: @@ -736,10 +736,13 @@ class BatchNormalizationBase(Layer):
  if self.virtual_batch_size is not None:
  # Virtual batches (aka ghost batches) can be simulated by reshaping the
  # Tensor and reusing the existing batch norm implementation
  - original_shape = [
  - d if d is not None else -1 for d in inputs.shape.as_list()]
  - original_shape = [-1] + original_shape[1:]
  - expanded_shape = [self.virtual_batch_size, -1] + original_shape[1:]
  + original_shape = array_ops.shape(inputs)
  + original_shape = array_ops.concat([
  + constant_op.constant([-1]),
  + original_shape[1:]], axis=0)
  + expanded_shape = array_ops.concat([
  + constant_op.constant([self.virtual_batch_size, -1]),
  + original_shape[1:]], axis=0)
  # Will cause errors if virtual_batch_size does not divide the batch size
  inputs = array_ops.reshape(inputs, expanded_shape)
  