code: def assert_shape(out1, out2):
      if out1.shape != out2.shape:
          msg = "Output shapes {} and {} don't match"
          raise AssertionError(msg.format(out1.shape, out2.shape))
  
imports: import os
  import numpy
  import oneflow
  from flowvision.models.alexnet import alexnet
  from flowvision.models.squeezenet import squeezenet1_0
  from flowvision.models.shufflenet_v2 import shufflenet_v2_x0_5
  from flowvision.models.mobilenet import mobilenet_v2
  from flowvision.models.ghostnet import ghostnet
  from flowvision.models.vision_transformer import vit_base_patch16_224
  import tvm
  import tvm.testing
  import tvm.topi.testing
  from tvm import relay
label_desc: Use image node_id 3 for calling a global function with example usage: assert_shape(out1, out2) without return types