code: def _pad_norm_assert_noshape(outputs):
      """Assert Masked BN/GN w/o spatial shape returns None shape."""
      outputs, spatial_shape = outputs
      assert spatial_shape is None
      return outputs, []
  
imports: import dataclasses
  from typing import Any, Callable, Dict, Sequence, Tuple, Type, Union
  from absl.testing import absltest
  from absl.testing import parameterized
  import flax.linen
  from jax import random
  import jax.numpy
  import numpy
  from scenic.model_lib.layers import masked_layers
label_desc: Use image node_id 2 for calling a global function with example usage: _pad_norm_assert_noshape(outputs) and returns: outputs,  list