





'''simple docstring'''






from torch import nn







def 					_a						(       _lowerCamelCase ) ->				Tuple:



			"""simple docstring"""


			if act_fn in ["swish", "silu"]:
						return nn.SiLU()
			elif act_fn == "mish":
						return nn.Mish()
			elif act_fn == "gelu":
						return nn.GELU()
			else:
						raise ValueError(F'''Unsupported activation function: {act_fn}''' )









'''simple docstring'''






def 					_a						(       _lowerCamelCase = 100 ) ->				int:



			"""simple docstring"""


			__snake_case					:      Any							      =     n * (n + 1) * (2 * n + 1) / 6
			__snake_case					:      List[Any]							      =     (n * (n + 1) / 2) ** 2
			return int(square_of_sum - sum_of_squares )


if __name__ == "__main__":
	print(f"""{solution() = }""")


