






def       snake_case_		(     lowerCAmelCase_       :						float       ,       lowerCAmelCase_       :						float      ):
							if mass < 0:
														raise ValueError("""The mass of a body cannot be negative"""      )
							return 0.5 * mass * abs(lowercase_      ) * abs(lowercase_      )


if __name__ == "__main__":
	import doctest

	doctest.testmod(verbose=True)













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