parameters: (self, hidden_size)
docstring: Sets the hidden size of the attention.
          Must be a multiple of the head_count.
docstring_summary: Sets the hidden size of the attention.
          Must be a multiple of the head_count.
docstring_tokens: Sets the hidden size of the attention . Must be a multiple of the head_count .
function: def hidden_size(self, hidden_size):
          """Sets the hidden size of the attention.
          Must be a multiple of the head_count.
          """
          if hidden_size < 1:
              raise ValueError('The `hidden_size` must be > 0.')
          self._internal.set_hidden_size(int(hidden_size))
function_tokens: def hidden_size ( self , hidden_size ) : if hidden_size < 1 : raise ValueError ( 'The `hidden_size` must be > 0.' ) self . _internal . set_hidden_size ( int ( hidden_size ) )