feat: add torch type declaration

crates/erg_compiler/lib/external/torch.d/__init__.d.er

@@ -0,0 +1 @@
+{.load!; .save!;} = pyimport "./serialization"

crates/erg_compiler/lib/external/torch.d/nn.d/__init__.d.er

@@ -0,0 +1 @@
+{.Module;} = pyimport "./modules/module"

crates/erg_compiler/lib/external/torch.d/nn.d/modules.d/__init__.d.er

@@ -0,0 +1 @@
+{.Module;} = pyimport "./module"

crates/erg_compiler/lib/external/torch.d/nn.d/modules.d/loss.d.er

@@ -0,0 +1,3 @@
+.CrossEntropyLoss: ClassType
+.CrossEntropyLoss.
+    __call__: () -> .CrossEntropyLoss

crates/erg_compiler/lib/external/torch.d/nn.d/modules.d/module.d.er

@@ -0,0 +1 @@
+.Module: ClassType

crates/erg_compiler/lib/external/torch.d/serialization.d.er

@@ -0,0 +1,2 @@
+.load!: (f: PathLike) => NoneType
+.save!: (obj: Obj, f: PathLike) => NoneType

crates/erg_compiler/lib/external/torch.d/utils.d/data.d/__init__.d.er

@@ -0,0 +1,2 @@
+{.DataLoader;} = pyimport "./dataloader"
+{.Dataset;} = pyimport "./dataset"

crates/erg_compiler/lib/external/torch.d/utils.d/data.d/dataloader.d.er

@@ -0,0 +1,9 @@
+dataset = pyimport "./dataset"
+
+.DataLoader: ClassType
+.DataLoader.
+    __call__: (
+        dataset: dataset.Dataset,
+        batch_size := Nat,
+        shuffle := Bool,
+    ) -> .DataLoader

crates/erg_compiler/lib/external/torch.d/utils.d/data.d/dataset.d.er

@@ -0,0 +1 @@
+.Dataset: ClassType

crates/erg_compiler/lib/external/torch.d/utils.d/optim.d/__init__.d.er

@@ -0,0 +1,28 @@
+.Optimizer: ClassType
+
+.ASGD: ClassType
+.ASGD <: .Optimizer
+.Adadelta: ClassType
+.Adadelta <: .Optimizer
+.Adagrad: ClassType
+.Adagrad <: .Optimizer
+.Adam: ClassType
+.Adam <: .Optimizer
+.AdamW: ClassType
+.AdamW <: .Optimizer
+.Adamax: ClassType
+.Adamax <: .Optimizer
+.LBFGS: ClassType
+.LBFGS <: .Optimizer
+.NAdam: ClassType
+.NAdam <: .Optimizer
+.RAdam: ClassType
+.RAdam <: .Optimizer
+.RMSprop: ClassType
+.RMSprop <: .Optimizer
+.Rprop: ClassType
+.Rprop <: .Optimizer
+.SGD: ClassType
+.SGD <: .Optimizer
+.SparseAdam: ClassType
+.SparseAdam <: .Optimizer

crates/erg_compiler/lib/external/torchvision.d/datasets.d/mnist.d.er

@@ -0,0 +1,23 @@
+vision = pyimport "./vision"
+
+.MNIST: ClassType
+.MNIST <: vision.VisionDataset
+.MNIST.
+    __call__: (
+        root: Str,
+        train := Bool,
+        download := Bool,
+        transform := GenericCallable,
+        target_transform := GenericCallable,
+    ) -> .MNIST
+
+.FashionMNIST: ClassType
+.FashionMNIST <: .MNIST
+.FashionMNIST.
+    __call__: (
+        root: Str,
+        train := Bool,
+        download := Bool,
+        transform := GenericCallable,
+        target_transform := GenericCallable,
+    ) -> .FashionMNIST

crates/erg_compiler/lib/external/torchvision.d/datasets.d/vision.d.er

@@ -0,0 +1,9 @@
+.VisionDataset: ClassType
+.VisionDataset.
+    __call__: (
+        root: Str,
+        train := Bool,
+        download := Bool,
+        transform := GenericCallable,
+        target_transform := GenericCallable,
+    ) -> .VisionDataset

crates/erg_compiler/lib/external/torchvision.d/transforms.d/transforms.d.er

@@ -0,0 +1,3 @@
+.ToTensor: ClassType
+.ToTensor.
+    __call__: () -> .ToTensor