



'''simple docstring'''
from __future__ import annotations






def A__    (     UpperCAmelCase_							):
						return len(set(UpperCAmelCase_							)							) == len(UpperCAmelCase_							)


if __name__ == "__main__":
					import doctest

					doctest.testmod()







from torch import nn


class       a							(			nn.Module				):



		def __init__(     self     :Tuple   ,__lowercase     :Optional[int]   ,__lowercase     :int  ):
						super().__init__()
						snake_case__	:      Optional[Any]     		=	class_size
						snake_case__	:      Dict     		=	embed_size
						# self.mlp1 = nn.Linear(embed_size, embed_size)
						# self.mlp2 = (nn.Linear(embed_size, class_size))
						snake_case__	:      Dict     		=	nn.Linear(__lowercase   ,__lowercase  )






		def 	__lowerCamelCase			(     self     :str   ,__lowercase     :int  ):
						# hidden_state = nn.functional.relu(self.mlp1(hidden_state))
						# hidden_state = self.mlp2(hidden_state)
						snake_case__	:      Optional[Any]     		=	self.mlp(__lowercase  )
						return logits

