v0.0.9

Author: BigNoobWasTaken

setup.py

@@ -7,7 +7,7 @@
 from setuptools import setup, find_packages
 from pybind11.setup_helpers import Pybind11Extension, build_ext
 
-__version__ = "0.0.8"
+__version__ = "0.0.9"
 
 def main():
     cwd = os.path.dirname(os.path.abspath(__file__))
@@ -18,7 +18,7 @@ def main():
 
     ext_modules = [
         Pybind11Extension(
-            "tensor_array._ext",
+            "tensor_array.tensor2",
             sources = glob.glob(os.path.join("cpp", "*.cc")),
             include_dirs=[tensor_array_lib_path + "/include"],
             library_dirs=[tensor_array_lib_path + "/lib/tensor-array", tensor_array_lib_path + "/lib64/tensor-array"],

src/tensor_array/core/constants.py

@@ -17,7 +17,7 @@ def zeros(shape : tuple, dtype : DataTypes = DataTypes.S_INT_32) -> Tensor:
         Tensor: A tensor filled with zeros.
     """
 
-    from .._ext.tensor2 import zeros as _zeros
+    from ..tensor2 import zeros as _zeros
     return _zeros(shape, dtype)
 
 def rand(shape : tuple, seed: int = 0) -> Tensor:
@@ -32,5 +32,5 @@ def rand(shape : tuple, seed: int = 0) -> Tensor:
         Tensor: A tensor filled with random values.
     """
 
-    from .._ext.tensor2 import rand as _rand
+    from ..tensor2 import rand as _rand
     return _rand(shape, seed)

src/tensor_array/core/datatypes.py

@@ -4,7 +4,7 @@
 # The DataTypes enum includes types such as BOOL, INT, FLOAT, DOUBLE, and others, which correspond to the data types used in tensors.
 """
 
-from .._ext.tensor2 import DataType as _DataType
+from ..tensor2 import DataType as _DataType
 from enum import Enum
 
 class DataTypes(Enum):

src/tensor_array/core/operator.py

@@ -15,7 +15,7 @@ def add(value_1 : Tensor, value_2 : Tensor) -> Tensor:
     Returns:
         Tensor: A tensor that is the element-wise sum of value_1 and value_2
     """
-    from .tensor2 import add as _add
+    from ..tensor2 import add as _add
     return _add(value_1, value_2)
 
 def divide(value_1 : Tensor, value_2 : Tensor) -> Tensor:
@@ -27,7 +27,7 @@ def divide(value_1 : Tensor, value_2 : Tensor) -> Tensor:
     Returns:
         Tensor: A tensor that is the element-wise division of value_1 by value_2
     """
-    from .tensor2 import divide as _divide
+    from ..tensor2 import divide as _divide
     return _divide(value_1, value_2)
 
 def multiply(value_1 : Tensor, value_2 : Tensor) -> Tensor:
@@ -39,7 +39,7 @@ def multiply(value_1 : Tensor, value_2 : Tensor) -> Tensor:
     Returns:
         Tensor: A tensor that is the element-wise product of value_1 and value_2
     """
-    from .tensor2 import multiply as _multiply
+    from ..tensor2 import multiply as _multiply
     return _multiply(value_1, value_2)
 
 def power(value_1 : Tensor, value_2 : Tensor) -> Tensor:
@@ -51,7 +51,7 @@ def power(value_1 : Tensor, value_2 : Tensor) -> Tensor:
     Returns:
         Tensor: A tensor that is the element-wise result of value_1 raised to the power of value_2
     """
-    from .tensor2 import power as _power
+    from ..tensor2 import power as _power
     return _power(value_1, value_2)
 
 def matmul(value_1 : Tensor, value_2 : Tensor) -> Tensor:
@@ -63,7 +63,7 @@ def matmul(value_1 : Tensor, value_2 : Tensor) -> Tensor:
     Returns:
         Tensor: A tensor that is the result of matrix multiplication between value_1 and value_2
     """
-    from .._ext.tensor2 import matmul as _matmul
+    from ..tensor2 import matmul as _matmul
     return _matmul(value_1, value_2)
 
 def condition(condition_value : Tensor, value_if_true : Tensor, value_if_false : Tensor) -> Tensor:
@@ -76,5 +76,5 @@ def condition(condition_value : Tensor, value_if_true : Tensor, value_if_false :
     Returns:
         Tensor: A tensor that is either value_if_true or value_if_false, depending on the condition.
     """
-    from .._ext.tensor2 import condition as _condition
+    from ..tensor2 import condition as _condition
     return _condition(condition_value, value_if_true, value_if_false)

src/tensor_array/core/tensor.py

@@ -5,7 +5,7 @@
 """
 
 from __future__ import annotations
-from .._ext.tensor2 import Tensor as _Tensor
+from ..tensor2 import Tensor as _Tensor
 from .datatypes import DataTypes
 
 class Tensor(_Tensor):