added stub files and bindings

Author: mbahng

aten/bindings/Tensor/gradtensor_binding.cpp

@@ -51,15 +51,15 @@ void init_gradtensor_binding(py::module_ &m) {
         py::arg("pidx") = 1
       )
     .def_static("ones_like", &GradTensor::ones_like, 
-        py::arg("shape")
+        py::arg("input")
       )
     .def_static("zeros", &GradTensor::zeros, 
         py::arg("shape") = std::vector<size_t>{1, 1},
         py::arg("bidx") = 0,  
         py::arg("pidx") = 1 
       )
     .def_static("zeros_like", &GradTensor::zeros_like, 
-        py::arg("shape")
+        py::arg("input")
       )
 
     // string 

ember/aten/__init__.pyi

@@ -59,7 +59,12 @@ class GradTensor(BaseTensor):
     bidx: int = 0, 
     pidx: int = 1
   ) -> 'GradTensor': ...
-
+  @staticmethod
+  def gaussian_like(
+    input: 'GradTensor',
+    mean: float = 0.0, 
+    stddev: float = 1.0
+  ) -> 'GradTensor': ...
   @staticmethod
   def uniform(
     shape: List[int] = [1, 1], 
@@ -68,20 +73,32 @@ class GradTensor(BaseTensor):
     bidx: int = 0, 
     pidx: int = 1
   ) -> 'GradTensor': ...
-
+  @staticmethod
+  def uniform_like(
+    input: 'GradTensor', 
+    min: float = 0.0, 
+    max: float = 1.0
+  ) -> 'GradTensor': ...
   @staticmethod
   def ones(
     shape: List[int] = [1, 1],
     bidx: int = 0, 
     pidx: int = 1
   ) -> 'GradTensor': ...
-
+  @staticmethod
+  def ones_like(
+    input: 'GradTensor'
+  ) -> 'GradTensor': ...
   @staticmethod
   def zeros(
     shape: List[int] = [1, 1],
     bidx: int = 0, 
     pidx: int = 1
   ) -> 'GradTensor': ...
+  @staticmethod
+  def zeros_like(
+    input: 'GradTensor'
+  ) -> 'GradTensor': ...
 
   # string
   def __str__(self) -> str: ...
@@ -210,23 +227,21 @@ class Tensor(BaseTensor):
     bidx: int = 0, 
     requires_grad: bool = True
   ) -> None: ...
-  
+
   @staticmethod
   def arange(
     start: int, 
     stop: int, 
     step: int = 1,
     requires_grad: bool = True
   ) -> 'Tensor': ...
-
   @staticmethod
   def linspace(
     start: float, 
     stop: float, 
     numsteps: int, 
     requires_grad: bool = True
   ) -> 'Tensor': ...
-
   @staticmethod
   def gaussian(
     shape: List[int], 
@@ -235,7 +250,12 @@ class Tensor(BaseTensor):
     bidx: int = 0, 
     requires_grad: bool = True
   ) -> 'Tensor': ...
-
+  @staticmethod
+  def gaussian_like(
+    input: 'Tensor', 
+    mean: float = 0.0, 
+    stddev: float = 1.0
+  ) -> 'Tensor': ...
   @staticmethod
   def uniform(
     shape: List[int], 
@@ -244,20 +264,32 @@ class Tensor(BaseTensor):
     bidx: int = 0, 
     requires_grad: bool = True
   ) -> 'Tensor': ...
-
+  @staticmethod
+  def uniform_like(
+    input: 'Tensor',
+    min: float = 0.0, 
+    max: float = 1.0
+  ) -> 'Tensor': ...
   @staticmethod
   def ones(
     shape: List[int],
     bidx: int = 0, 
     requires_grad: bool = True
   ) -> 'Tensor': ...
-
+  @staticmethod
+  def ones_like(
+    input: 'Tensor'
+  ) -> 'Tensor': ...
   @staticmethod
   def zeros(
     shape: List[int],
     bidx: int = 0, 
     requires_grad: bool = True
   ) -> 'Tensor': ...
+  @staticmethod
+  def zeros_like(
+    input: 'Tensor'
+  ) -> 'Tensor': ...
 
   # string
   def __str__(self) -> str: ...