{".class": "MypyFile", "_fullname": "torch.nn.modules.sparse", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Embedding": {".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.modules.sparse.Embedding", "name": "Embedding", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.sparse.Embedding", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.sparse", "mro": ["torch.nn.modules.sparse.Embedding", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__constants__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.sparse.Embedding.__constants__", "name": "__constants__", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "num_embeddings", "embedding_dim", "padding_idx", "max_norm", "norm_type", "scale_grad_by_freq", "sparse", "_weight", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.sparse.Embedding.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "num_embeddings", "embedding_dim", "padding_idx", "max_norm", "norm_type", "scale_grad_by_freq", "sparse", "_weight", "device", "dtype"], "arg_types": ["torch.nn.modules.sparse.Embedding", "builtins.int", "builtins.int", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".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 Embedding", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "_fill_padding_idx_with_zero": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.sparse.Embedding._fill_padding_idx_with_zero", "name": "_fill_padding_idx_with_zero", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.sparse.Embedding"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_fill_padding_idx_with_zero of Embedding", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "embedding_dim": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.embedding_dim", "name": "embedding_dim", "type": "builtins.int"}}, "extra_repr": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.sparse.Embedding.extra_repr", "name": "extra_repr", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.sparse.Embedding"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "extra_repr of Embedding", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "flags": [], "fullname": "torch.nn.modules.sparse.Embedding.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "arg_types": ["torch.nn.modules.sparse.Embedding", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of Embedding", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "from_pretrained": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["cls", "embeddings", "freeze", "padding_idx", "max_norm", "norm_type", "scale_grad_by_freq", "sparse"], "flags": ["is_class", "is_decorated"], "fullname": "torch.nn.modules.sparse.Embedding.from_pretrained", "name": "from_pretrained", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_classmethod", "is_ready"], "fullname": null, "name": "from_pretrained", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["cls", "embeddings", "freeze", "padding_idx", "max_norm", "norm_type", "scale_grad_by_freq", "sparse"], "arg_types": [{".class": "TypeType", "item": "torch.nn.modules.sparse.Embedding"}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": true, "is_ellipsis_args": false, "name": "from_pretrained of Embedding", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}}, "max_norm": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.max_norm", "name": "max_norm", "type": {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}}}, "norm_type": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.norm_type", "name": "norm_type", "type": "builtins.float"}}, "num_embeddings": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.num_embeddings", "name": "num_embeddings", "type": "builtins.int"}}, "padding_idx": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.padding_idx", "name": "padding_idx", "type": {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.sparse.Embedding.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.sparse.Embedding"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of Embedding", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "scale_grad_by_freq": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.scale_grad_by_freq", "name": "scale_grad_by_freq", "type": "builtins.bool"}}, "sparse": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.sparse", "name": "sparse", "type": "builtins.bool"}}, "weight": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.Embedding.weight", "name": "weight", "type": "torch._tensor.Tensor"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "EmbeddingBag": {".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.modules.sparse.EmbeddingBag", "name": "EmbeddingBag", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.sparse.EmbeddingBag", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.sparse", "mro": ["torch.nn.modules.sparse.EmbeddingBag", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__constants__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.__constants__", "name": "__constants__", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "num_embeddings", "embedding_dim", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "_weight", "include_last_offset", "padding_idx", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.sparse.EmbeddingBag.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "num_embeddings", "embedding_dim", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "_weight", "include_last_offset", "padding_idx", "device", "dtype"], "arg_types": ["torch.nn.modules.sparse.EmbeddingBag", "builtins.int", "builtins.int", {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.str", "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".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 EmbeddingBag", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "_fill_padding_idx_with_zero": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.sparse.EmbeddingBag._fill_padding_idx_with_zero", "name": "_fill_padding_idx_with_zero", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.sparse.EmbeddingBag"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_fill_padding_idx_with_zero of EmbeddingBag", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "embedding_dim": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.embedding_dim", "name": "embedding_dim", "type": "builtins.int"}}, "extra_repr": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.sparse.EmbeddingBag.extra_repr", "name": "extra_repr", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.sparse.EmbeddingBag"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "extra_repr of EmbeddingBag", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["self", "input", "offsets", "per_sample_weights"], "flags": [], "fullname": "torch.nn.modules.sparse.EmbeddingBag.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1], "arg_names": ["self", "input", "offsets", "per_sample_weights"], "arg_types": ["torch.nn.modules.sparse.EmbeddingBag", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of EmbeddingBag", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "from_pretrained": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["cls", "embeddings", "freeze", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "include_last_offset", "padding_idx"], "flags": ["is_class", "is_decorated"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.from_pretrained", "name": "from_pretrained", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["cls", "embeddings", "freeze", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "include_last_offset", "padding_idx"], "arg_types": [{".class": "TypeType", "item": "torch.nn.modules.sparse.EmbeddingBag"}, "torch._tensor.Tensor", "builtins.bool", {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.str", "builtins.bool", "builtins.bool", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "cls"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "from_pretrained of EmbeddingBag", "ret_type": "torch.nn.modules.sparse.EmbeddingBag", "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_classmethod", "is_ready"], "fullname": null, "name": "from_pretrained", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["cls", "embeddings", "freeze", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "include_last_offset", "padding_idx"], "arg_types": [{".class": "TypeType", "item": "torch.nn.modules.sparse.EmbeddingBag"}, "torch._tensor.Tensor", "builtins.bool", {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.str", "builtins.bool", "builtins.bool", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "cls"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "from_pretrained of EmbeddingBag", "ret_type": "torch.nn.modules.sparse.EmbeddingBag", "type_guard": null, "variables": []}}}}, "include_last_offset": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.include_last_offset", "name": "include_last_offset", "type": "builtins.bool"}}, "max_norm": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.max_norm", "name": "max_norm", "type": {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}}}, "mode": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.mode", "name": "mode", "type": "builtins.str"}}, "norm_type": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.norm_type", "name": "norm_type", "type": "builtins.float"}}, "num_embeddings": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.num_embeddings", "name": "num_embeddings", "type": "builtins.int"}}, "padding_idx": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.padding_idx", "name": "padding_idx", "type": {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.sparse.EmbeddingBag.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.sparse.EmbeddingBag"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of EmbeddingBag", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "scale_grad_by_freq": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.scale_grad_by_freq", "name": "scale_grad_by_freq", "type": "builtins.bool"}}, "sparse": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.sparse", "name": "sparse", "type": "builtins.bool"}}, "weight": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.sparse.EmbeddingBag.weight", "name": "weight", "type": "torch._tensor.Tensor"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "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"}, "Parameter": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parameter.Parameter", "kind": "Gdef"}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.sparse.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.sparse.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.sparse.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.sparse.__package__", "name": "__package__", "type": "builtins.str"}}, "init": {".class": "SymbolTableNode", "cross_ref": "torch.nn.init", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/modules/sparse.py"}