code: def test_multi_head_restore_head_size():
      block = blocks.basic.MultiHeadSelfAttention(head_size=16)
  
      block = blocks.basic.MultiHeadSelfAttention.from_config(
          block.get_config()
      )
  
      assert block.head_size == 16
  
imports: import keras_tuner
  import pytest
  import tensorflow
  from tensorflow import keras
  from tensorflow import nest
  from autokeras import blocks
  from autokeras import test_utils
label_desc: Use image node_id 37 for calling a global function with example usage: test_multi_head_restore_head_size() without return types