code: def test_fully_connected_head(self):
      batch_size = 2
      in_plane = 3
      image_size = 4
      num_classes = 5
      head = FullyConnectedHead(
          "default_head",
          num_classes=num_classes,
          in_plane=in_plane,
      )
      input = torch.rand([batch_size, in_plane, image_size, image_size])
      output = head(input)
      self.assertEqual(
          output.shape, torch.Size([batch_size, num_classes])
      )
  
file_path: ClassyVision/test/heads_fully_connected_head_test.py
imports: import copy
  from functools import partial
  import torch
  from classy_vision.generic.util import get_torch_version
  from classy_vision.heads import build_head
  from classy_vision.heads.fully_connected_head import FullyConnectedHead
  from test.generic.utils import ClassyTestCase
label_desc: Use image node_id 1 for calling the TestFullyConnectedHead obj's underlying member method code with example usage: obj.test_fully_connected_head() without return types