



'''simple docstring'''






from typing import TYPE_CHECKING

from ...utils import _LazyModule


A_							: int						 =	{"""tokenization_bertweet""": ["""BertweetTokenizer"""]}


if TYPE_CHECKING:
   from .tokenization_bertweet import BertweetTokenizer

else:
   import sys

   A_							: int						 =	_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)








'''simple docstring'''






from torch import nn



class 							lowercase       (  nn.Module				):


 """simple docstring"""


 def __init__(							self  ,a_  ,a_							)		->   List[Any]:
        super().__init__()
        _UpperCAmelCase  :						Dict			     = class_size
        _UpperCAmelCase  :						Union[str, Any]			     = embed_size
        # self.mlp1 = nn.Linear(embed_size, embed_size)
        # self.mlp2 = (nn.Linear(embed_size, class_size))
        _UpperCAmelCase  :						List[Any]			     = nn.Linear(a_  ,a_							)







 def _snake_case (							self  ,a_							)		->   Tuple:
        # hidden_state = nn.functional.relu(self.mlp1(hidden_state))
        # hidden_state = self.mlp2(hidden_state)
        _UpperCAmelCase  :						Optional[int]			     = self.mlp(a_							)
        return logits



