code: def test_fusion_different_dimensions():
      from mxnet.gluon import HybridBlock
  
      class Foo(HybridBlock):
          def __init__(self):
              super(Foo, self).__init__()
  
          def forward(self, x):
              mask2 = x.astype(np.float32)
              mask = mx.np.expand_dims(mask2, axis=-1)
              return mask
  
      foo = Foo()
      foo.hybridize(static_alloc=True)
      # Pass 1-D data
      out = foo(mx.np.ones((10,), ctx=mx.gpu()))
      assert np.all(out.asnumpy() == np.ones((10, 1)))
      assert out.shape == (10, 1)
      # Pass 2-D data
      out = foo(mx.np.ones((10, 10), ctx=mx.gpu()))
      assert np.all(out.asnumpy() == np.ones((10, 10)))
      assert out.shape == (10, 10, 1)
  
file_path: mxnet/tests/python/gpu/test_fusion.py
imports: import sys
  import os
  import random
  import itertools
  import mxnet
  import numpy
  from mxnet import autograd, gluon
  from mxnet.test_utils import *
label_desc: Use image node_id 9 for calling a global function with example usage: test_fusion_different_dimensions() and returns: mask