code: def _check_shape(arr):
      """
      return True if shape is of form
      (1,C,1,1) or (C,1,1)
      """
      rank = len(arr.shape)
      if not (rank == 3 or rank == 4):
          return False
      C = arr.shape[-3]
      if not (arr.shape == (1, C, 1, 1) or arr.shape == (C, 1, 1)):
          return False
      return True
  
file_path: turicreate/src/external/coremltools_wrap/coremltools/coremltools/converters/mil/mil/passes/elementwise_batchnorm_fusion.py
imports: from __future__ import print_function
  from __future__ import division
  from __future__ import absolute_import
  from coremltools.converters.mil.mil.passes.pass_registry import register_pass
  from coremltools.converters.mil.mil import Builder
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: _check_shape(arr) and returns: True, False, False