code: def get_random_axis(data: relay.Expr):
      rank = len(relay.transform.InferTypeLocal(data).shape)
      return random.randint(0, rank - 1)
  
imports: import itertools
  import random
  import tempfile
  from typing import Callable, Dict, List, Optional, Tuple, Union
  import numpy
  import pytest
  import tvm
  import tvm.testing
  from tvm import meta_schedule, relay
  from tvm.meta_schedule.schedule.cuda.layout_transform import cuda_layout_transform_schedule_rule
  from tvm.relay.op import OpPattern
  from tvm.script import ir
  from tvm.script import tir
  from tvm.tir.schedule import BlockRV
label_desc: Use image node_id 1 for calling a global function with example usage: get_random_axis(data) and returns: random