testing

Author: BigNoobWasTaken

src/main.py

@@ -1,8 +1,8 @@
-from tensor_array.core import tensor2 as t
+from tensor_array.core import Tensor
 
 print("hello")
 
-t1 = t.Tensor([[1, 2, 3], [4, 5, 6]])
+t1 = Tensor([[1, 2, 3], [4, 5, 6]])
 t2 = t1.clone()
 print("tensor len", t1.__len__())
 print(t1)

src/tensor_array/core/__pycache__/__init__.cpython-310.pyc

@@ -1,12 +1,18 @@
 from tensor_array.layers import Layer
+from tensor_array.layers import Parameter
 from tensor_array.core import Tensor
 from typing import Any
 
 
 class Linear(Layer):
-    def __init__(self) -> None:
-        self.w = t.Tensor(0)
-        self.b = t.Tensor(0)
+    def __init__(self, bias) -> None:
+        super(self)
+        self.bias_shape = bias
+        self.b = Parameter(Tensor(shape = (bias)))
 
-    def __call__(self, input) -> Any:
-        return input @ self.w + self.b
+    def init_value(self, t):
+        self.w = Parameter(Tensor(shape = (t.shape(-1), self.bias_shape)))
+    
+    def calculate(self, t):
+        return self.w @ t + self.b
+