{".class": "MypyFile", "_fullname": "torch.nn.utils.parametrizations", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Enum": {".class": "SymbolTableNode", "cross_ref": "enum.Enum", "kind": "Gdef"}, "F": {".class": "SymbolTableNode", "cross_ref": "torch.nn.functional", "kind": "Gdef"}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef"}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef"}, "_OrthMaps": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["enum.Enum"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.utils.parametrizations._OrthMaps", "name": "_OrthMaps", "type_vars": []}, "flags": ["is_enum"], "fullname": "torch.nn.utils.parametrizations._OrthMaps", "metaclass_type": "enum.EnumMeta", "metadata": {}, "module_name": "torch.nn.utils.parametrizations", "mro": ["torch.nn.utils.parametrizations._OrthMaps", "enum.Enum", "builtins.object"], "names": {".class": "SymbolTable", "cayley": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.utils.parametrizations._OrthMaps.cayley", "name": "cayley", "type": "enum.auto"}}, "householder": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.utils.parametrizations._OrthMaps.householder", "name": "householder", "type": "enum.auto"}}, "matrix_exp": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.utils.parametrizations._OrthMaps.matrix_exp", "name": "matrix_exp", "type": "enum.auto"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "_Orthogonal": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.module.Module"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.utils.parametrizations._Orthogonal", "name": "_Orthogonal", "type_vars": []}, "flags": [], "fullname": "torch.nn.utils.parametrizations._Orthogonal", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.utils.parametrizations", "mro": ["torch.nn.utils.parametrizations._Orthogonal", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 5], "arg_names": ["self", "weight", "orthogonal_map", "use_trivialization"], "flags": [], "fullname": "torch.nn.utils.parametrizations._Orthogonal.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 5], "arg_names": ["self", "weight", "orthogonal_map", "use_trivialization"], "arg_types": ["torch.nn.utils.parametrizations._Orthogonal", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "torch.nn.utils.parametrizations._OrthMaps", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of _Orthogonal", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "base": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.utils.parametrizations._Orthogonal.base", "name": "base", "type": "torch._tensor.Tensor"}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "X"], "flags": [], "fullname": "torch.nn.utils.parametrizations._Orthogonal.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "X"], "arg_types": ["torch.nn.utils.parametrizations._Orthogonal", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of _Orthogonal", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "orthogonal_map": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._Orthogonal.orthogonal_map", "name": "orthogonal_map", "type": "torch.nn.utils.parametrizations._OrthMaps"}}, "right_inverse": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "Q"], "flags": ["is_decorated"], "fullname": "torch.nn.utils.parametrizations._Orthogonal.right_inverse", "name": "right_inverse", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "Q"], "arg_types": ["torch.nn.utils.parametrizations._Orthogonal", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "right_inverse of _Orthogonal", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": null, "name": "right_inverse", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "Q"], "arg_types": ["torch.nn.utils.parametrizations._Orthogonal", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "right_inverse of _Orthogonal", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}}, "shape": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._Orthogonal.shape", "name": "shape", "type": {".class": "AnyType", "missing_import_name": null, "source_any": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_of_any": 7}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "_SpectralNorm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.module.Module"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.utils.parametrizations._SpectralNorm", "name": "_SpectralNorm", "type_vars": []}, "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.utils.parametrizations", "mro": ["torch.nn.utils.parametrizations._SpectralNorm", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["self", "weight", "n_power_iterations", "dim", "eps"], "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["self", "weight", "n_power_iterations", "dim", "eps"], "arg_types": ["torch.nn.utils.parametrizations._SpectralNorm", "torch._tensor.Tensor", "builtins.int", "builtins.int", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of _SpectralNorm", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "_power_method": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "weight_mat", "n_power_iterations"], "flags": ["is_decorated"], "fullname": "torch.nn.utils.parametrizations._SpectralNorm._power_method", "name": "_power_method", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "weight_mat", "n_power_iterations"], "arg_types": ["torch.nn.utils.parametrizations._SpectralNorm", "torch._tensor.Tensor", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_power_method of _SpectralNorm", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": null, "name": "_power_method", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "weight_mat", "n_power_iterations"], "arg_types": ["torch.nn.utils.parametrizations._SpectralNorm", "torch._tensor.Tensor", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_power_method of _SpectralNorm", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}}, "_reshape_weight_to_matrix": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "weight"], "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm._reshape_weight_to_matrix", "name": "_reshape_weight_to_matrix", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "weight"], "arg_types": ["torch.nn.utils.parametrizations._SpectralNorm", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_reshape_weight_to_matrix of _SpectralNorm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "_u": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm._u", "name": "_u", "type": "torch._tensor.Tensor"}}, "_v": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm._v", "name": "_v", "type": "torch._tensor.Tensor"}}, "dim": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm.dim", "name": "dim", "type": "builtins.int"}}, "eps": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm.eps", "name": "eps", "type": "builtins.float"}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "weight"], "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "weight"], "arg_types": ["torch.nn.utils.parametrizations._SpectralNorm", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of _SpectralNorm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "n_power_iterations": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm.n_power_iterations", "name": "n_power_iterations", "type": "builtins.int"}}, "right_inverse": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "value"], "flags": [], "fullname": "torch.nn.utils.parametrizations._SpectralNorm.right_inverse", "name": "right_inverse", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "value"], "arg_types": ["torch.nn.utils.parametrizations._SpectralNorm", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "right_inverse of _SpectralNorm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrizations.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrizations.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrizations.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrizations.__package__", "name": "__package__", "type": "builtins.str"}}, "_is_orthogonal": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["Q", "eps"], "flags": [], "fullname": "torch.nn.utils.parametrizations._is_orthogonal", "name": "_is_orthogonal", "type": null}}, "_make_orthogonal": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["A"], "flags": [], "fullname": "torch.nn.utils.parametrizations._make_orthogonal", "name": "_make_orthogonal", "type": null}}, "auto": {".class": "SymbolTableNode", "cross_ref": "enum.auto", "kind": "Gdef"}, "orthogonal": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 5], "arg_names": ["module", "name", "orthogonal_map", "use_trivialization"], "flags": [], "fullname": "torch.nn.utils.parametrizations.orthogonal", "name": "orthogonal", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 5], "arg_names": ["module", "name", "orthogonal_map", "use_trivialization"], "arg_types": ["torch.nn.modules.module.Module", "builtins.str", {".class": "UnionType", "items": ["builtins.str", {".class": "NoneType"}]}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "orthogonal", "ret_type": "torch.nn.modules.module.Module", "type_guard": null, "variables": []}}}, "parametrize": {".class": "SymbolTableNode", "cross_ref": "torch.nn.utils.parametrize", "kind": "Gdef"}, "spectral_norm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["module", "name", "n_power_iterations", "eps", "dim"], "flags": [], "fullname": "torch.nn.utils.parametrizations.spectral_norm", "name": "spectral_norm", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["module", "name", "n_power_iterations", "eps", "dim"], "arg_types": ["torch.nn.modules.module.Module", "builtins.str", "builtins.int", "builtins.float", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "spectral_norm", "ret_type": "torch.nn.modules.module.Module", "type_guard": null, "variables": []}}}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/utils/parametrizations.py"}