






from torch import nn

class 		lowerCAmelCase     ( nn.Module			):

					'''simple docstring'''

					def __init__(  self				:	Tuple		,     __a				:	int		,     __a				:	Optional[int]							)			->  Union[str, Any]:


												"""simple docstring"""

												super().__init__()
												__lowercase      :							int				=     class_size
												__lowercase      :							int				=     embed_size
												# self.mlp1 = nn.Linear(embed_size, embed_size)
												# self.mlp2 = (nn.Linear(embed_size, class_size))
												__lowercase      :							str				=     nn.Linear(__a		,     __a							)



					def       lowerCAmelCase				(  self				:	Tuple		,     __a				:	int							)			->  Tuple:


												"""simple docstring"""

												__lowercase      :							str				=     self.mlp(__a							)
												return logits







def       snake_case_		(     lowerCAmelCase_       :						int       ,       lowerCAmelCase_       :						int      ):
							return int((input_a, input_a).count(0      ) == 0      )
def       snake_case_		(     ):
							assert and_gate(0       ,       0      ) == 0
							assert and_gate(0       ,       1      ) == 0
							assert and_gate(1       ,       0      ) == 0
							assert and_gate(1       ,       1      ) == 1


if __name__ == "__main__":
	test_and_gate()
	print(and_gate(1, 0))
	print(and_gate(0, 0))
	print(and_gate(0, 1))
	print(and_gate(1, 1))