code: def test_clf_head_build_with_zero_dropout_return_tensor():
      block = head_module.ClassificationHead(dropout=0, shape=(8,))
  
      outputs = block.build(
          keras_tuner.HyperParameters(),
          keras.Input(shape=(5,), dtype=tf.float32),
      )
  
      assert len(nest.flatten(outputs)) == 1
  
file_path: autokeras/autokeras/blocks/heads_test.py
imports: import keras_tuner
  import numpy
  import tensorflow
  from tensorflow import keras
  from tensorflow import nest
  import autokeras
  from autokeras import hyper_preprocessors
  from autokeras import nodes
  from autokeras import preprocessors
  from autokeras import test_utils
  from autokeras.blocks import heads
label_desc: Use image node_id 6 for calling a global function with example usage: test_clf_head_build_with_zero_dropout_return_tensor() without return types