

"""simple docstring"""




from torch import nn







class 			lowercase    (	nn.Module      ):




				def __init__( self				,       snake_case				,       snake_case	):
											super().__init__()
											snake_case_          =   class_size
											snake_case_          =   embed_size
											# self.mlp1 = nn.Linear(embed_size, embed_size)
											# self.mlp2 = (nn.Linear(embed_size, class_size))
											snake_case_          =   nn.Linear(snake_case				,       snake_case	)




				def    a							( self				,       snake_case	):
											# hidden_state = nn.functional.relu(self.mlp1(hidden_state))
											# hidden_state = self.mlp2(hidden_state)
											snake_case_          =   self.mlp(snake_case	)
											return logits







from __future__ import annotations





def  __lowerCamelCase     (			UpperCamelCase__       ,							UpperCamelCase__				):







							'''simple docstring'''






							snake_case_          =   []
							create_all_state(1       ,							UpperCamelCase__       ,							UpperCamelCase__       ,							[]       ,							UpperCamelCase__				)
							return result





def  __lowerCamelCase     (			UpperCamelCase__       ,							UpperCamelCase__       ,							UpperCamelCase__       ,							UpperCamelCase__       ,							UpperCamelCase__       ,							):







							'''simple docstring'''






							if level == 0:
														total_list.append(current_list[:]				)
														return

							for i in range(UpperCamelCase__       ,							total_number - level + 2				):
														current_list.append(UpperCamelCase__				)
														create_all_state(i + 1       ,							UpperCamelCase__       ,							level - 1       ,							UpperCamelCase__       ,							UpperCamelCase__				)
														current_list.pop()





def  __lowerCamelCase     (			UpperCamelCase__				):







							'''simple docstring'''






							for i in total_list:
														print(*UpperCamelCase__				)


if __name__ == "__main__":
					_UpperCAmelCase		:       str						    =	4
					_UpperCAmelCase		:       Tuple						    =	2
					_UpperCAmelCase		:       Optional[int]						    =	generate_all_combinations(n, k)
					print_all_state(total_list)




