code: def testFCAxis2(self):
      model = model_helper.ModelHelper(name="test_model")
      model.net.FC(["x", "w", "b"], ["y"], axis=2)
      workspace.FeedBlob(
          "x", np.random.rand(4, 20, 36).astype(np.float32)
      )
      workspace.FeedBlob("w", np.random.rand(36, 36).astype(np.float32))
      workspace.FeedBlob(
          "b",
          np.random.rand(
              36,
          ).astype(np.float32),
      )
      self.InferTensorRunAndCompare(model)
  
imports: import numpy
  import unittest
  from caffe2.proto import caffe2_pb2
  from caffe2.python import core, workspace, test_util, model_helper, brew, build
label_desc: Use image node_id 2 for calling the TestShapeInference obj's underlying member method code with example usage: obj.testFCAxis2() without return types