code: def test_classification_head(
      self, input_shape, num_classes, momentum, training
  ):
      B = input_shape[0]
      inputs = tf.random.uniform(input_shape)
      head = ClassificationHead(num_classes, momentum)
      outputs = head(inputs, training=training)
      assert outputs.shape == (B, num_classes)
  
file_path: graphics/tensorflow_graphics/nn/layer/tests/pointnet_test.py
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_graphics.nn.layer.pointnet import ClassificationHead
  from tensorflow_graphics.nn.layer.pointnet import PointNetConv2Layer
  from tensorflow_graphics.nn.layer.pointnet import PointNetDenseLayer
  from tensorflow_graphics.nn.layer.pointnet import PointNetVanillaClassifier
  from tensorflow_graphics.nn.layer.pointnet import VanillaEncoder
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 4 for calling the RandomForwardExecutionTest obj's underlying member method code with example usage: obj.test_classification_head(input_shape, num_classes, momentum, training) without return types