Tensor-Array
diff --git a/‎include/tensor_bind.cc
Lines changed: 1 addition & 0 deletions b/‎include/tensor_bind.cc
Lines changed: 1 addition & 0 deletions
diff --git a/‎src/TensorArray/core/libTensorCore_Core.so renamed to ‎src/TensorArray/core/libTensorArray_Core.so
12.8 MB b/‎src/TensorArray/core/libTensorCore_Core.so renamed to ‎src/TensorArray/core/libTensorArray_Core.so
12.8 MB
diff --git a/‎src/TensorArray/core/tensor2.so
-720 Bytes b/‎src/TensorArray/core/tensor2.so
-720 Bytes
diff --git a/‎src/TensorArray/layers/Linear.py
Lines changed: 3 additions & 4 deletions b/‎src/TensorArray/layers/Linear.py
Lines changed: 3 additions & 4 deletions
@@ -170,6 +170,7 @@ PYBIND11_MODULE(tensor2, m)
 		.def(hash(pybind11::self))
 		.def("transpose", &Tensor::transpose)
 		.def("calc_grad", &Tensor::calc_grad)
+		.def("get_grad", &Tensor::get_grad)
 		.def("sin", &Tensor::sin)
 		.def("sin", &Tensor::sin)
 		.def("cos", &Tensor::cos)
 
@@ -1,12 +1,11 @@
-
+from TensorArray.core import tensor2 as t
 from typing import Any
 
 
 class Linear:
     def __init__(self) -> None:
-        self.w
-        self.b
-        pass
+        self.w = t.Tensor(0)
+        self.b = t.Tensor(0)
 
     def __call__(self, input) -> Any:
         return input @ self.w + self.b