Skip to content

Commit a3ea643

Browse files
fixed bugs
1 parent d9d5808 commit a3ea643

File tree

10 files changed

+13
-6
lines changed

10 files changed

+13
-6
lines changed
-11.9 MB
Binary file not shown.

src/TensorArray/core/tensor2.so

-4 MB
Binary file not shown.

src/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from TensorArray.core import tensor2 as t
1+
from tensor_array.core import tensor2 as t
22

33
print("hello")
44

src/tensor_array/core/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from tensor_array.core.tensor2 import Tensor

src/tensor_array/core/tensor2.so

1.37 KB
Binary file not shown.

src/tensor_array/layers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
from layer import Layer
2+
from parameter import Parameter

src/tensor_array/util/layer.py renamed to src/tensor_array/layers/layer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,9 @@
55
from .parameter import Parameter
66

77
class Layer:
8+
"""
9+
Layer class
10+
"""
811
is_running: bool
912
_layers: Dict[str, Optional['Layer']]
1013
_parameters: Dict[str, Optional[Parameter]]
@@ -17,8 +20,9 @@ def __init__(self) -> None:
1720
super().__setattr__('_tensors', OrderedDict())
1821

1922
def __call__(self, *args: Any, **kwds: Any) -> Any:
20-
if not self.__dict__.get('is_running'):
23+
if not super().__getattr__('is_running'):
2124
self.init_value(args, kwds)
25+
super().__setattr__('is_running', True)
2226
self.calculate(args, kwds)
2327

2428
def init_value(self, *args: Any, **kwds: Any) -> Any:

src/tensor_array/util/parameter.py renamed to src/tensor_array/layers/parameter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from ..core.tensor2 import tensor2 as t
1+
from tensor_array.core.tensor2 import tensor2 as t
22

33
class Parameter:
44
def __init__(self, tensor_param: t.Tensor) -> None:

src/tensor_array/layers/linear.py renamed to src/tensor_array/layers/util/linear.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
1-
from TensorArray.core import tensor2 as t
1+
from tensor_array.layers import Layer
2+
from tensor_array.core import Tensor
23
from typing import Any
34

45

5-
class Linear:
6+
class Linear(Layer):
67
def __init__(self) -> None:
78
self.w = t.Tensor(0)
89
self.b = t.Tensor(0)

src/tensor_array/util/__init__.py

Lines changed: 0 additions & 1 deletion
This file was deleted.

0 commit comments

Comments
 (0)