Module ImportFrom alias ImportFrom alias identifier FunctionDef arguments args NameParam NameParam defaults body Assert CompareGt NameLoad Num Str Assign NameStore Call AttributeLoad NameLoad attr For NameStore Call NameLoad BinOpSub NameLoad Num body Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr Return NameLoad decorator_list Assign NameStore NameLoad FunctionDef arguments args NameParam defaults body Return SubscriptLoad NameLoad Index Num decorator_list FunctionDef arguments args NameParam defaults body Return SubscriptLoad NameLoad Index Num decorator_list FunctionDef arguments args NameParam defaults body Return Call AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Return Call AttributeLoad NameLoad attr decorator_list ClassDef bases NameLoad body FunctionDef arguments args NameParam NameParam NameParam defaults Str body Expr Call AttributeLoad Call NameLoad NameLoad NameLoad attr NameLoad Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body Assert Call NameLoad NameLoad NameLoad Return TupleLoad decorator_list FunctionDef arguments args NameParam NameParam defaults kwarg body Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad Assert Call NameLoad Call AttributeLoad NameLoad attr NameLoad If CompareEq AttributeLoad NameLoad attr Str body Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr Num keyword Num orelse Raise Call NameLoad AttributeLoad NameLoad attr Return Call AttributeLoad NameLoad attr NameLoad TupleLoad decorator_list decorator_list
N/A node NestedNode N/A backend T Repeat N/A N/A node n_repeats N/A N/A N/A N/A n_repeats 0 Must repeat a positive number of times N/A new_node N/A N/A node copy N/A i N/A range N/A n_repeats 1 N/A N/A new_node N/A N/A new_node chain N/A N/A node copy N/A new_node N/A N/A MLP Repeat Last N/A N/A node N/A N/A N/A N/A node N/A -1 N/A First N/A N/A node N/A N/A N/A N/A node N/A 0 N/A Freeze N/A N/A node N/A N/A N/A N/A N/A node freeze N/A Unfreeze N/A N/A node N/A N/A N/A N/A N/A node unfreeze N/A Norm N/A NestedNode N/A __init__ N/A N/A self node type N/A l2 N/A N/A N/A N/A N/A super Norm self __init__ node N/A N/A self type type N/A _infer N/A N/A self shape_in N/A N/A N/A N/A isinstance shape_in int N/A N/A N/A forward N/A N/A self X N/A kwargs N/A N/A X N/A N/A N/A self node forward X kwargs N/A N/A isinstance N/A N/A X get_shape_out int N/A N/A N/A self type l2 N/A N/A out N/A N/A T sum N/A N/A T pow N/A N/A X get_data 2 axis -1 N/A N/A N/A NotImplementedError N/A self type N/A N/A N/A X next out N/A N/A N/A