[Feat] 新增 nn.Module() 以及 删除无法使用特性

.github/workflows/autopackage_linux_x86_64_cpu.yml

@@ -1,16 +1,9 @@
 name: Publish Axono Package (Linux x86_64 For CPU)
 
-# on:
-#   push:
-#     tags:
-#       - 'v*'
-
 on:
   push:
-    branches: [ dev ]
-  pull_request:
-    branches: [ dev ]
-  workflow_dispatch:  # 添加手动触发器
+    tags:
+      - 'v*'
 
 jobs:
   build:

.github/workflows/autopackage_linux_x86_64_cuda.yml

@@ -1,16 +1,9 @@
 name: Publish Axono Package (Linux x86_64 For CUDA)
 
-# on:
-#  push:
-#    tags:
-#      - 'v*'
-
 on:
-  push:
-    branches: [ dev ]
-  pull_request:
-    branches: [ dev ]
-  workflow_dispatch:  # 添加手动触发器
+ push:
+   tags:
+     - 'v*'
 
 jobs:
   build:

include/axono/core/module.h

@@ -0,0 +1,16 @@
+#include "tensor.h"
+
+namespace axono::core {
+class Module {
+private:
+    std::unordered_map<std::string, Tensor> weights_;  // 存储权重张量
+public:
+    void add_weight(const std::string& name, const Tensor& weight) {
+        weights_[name] = weight;
+    }
+    Tensor& get_weight(const std::string& name) {
+        return weights_.at(name);
+    }
+    auto& weights() { return weights_; }
+};
+}

include/axono/pybind/core/module.h

@@ -0,0 +1,23 @@
+#include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
+#include "axono/core/module.h"
+
+namespace py = pybind11;
+
+void init_module(py::module &m) {
+    py::class_<axono::core::Module>(m, "Module")
+        .def(py::init<>(), "创建一个空的 Module 实例")
+        .def("add_weight", 
+             &axono::core::Module::add_weight, 
+             py::arg("name"), py::arg("weight"), 
+             "向模块添加权重张量")
+        .def("get_weight", 
+             &axono::core::Module::get_weight, 
+             py::arg("name"), 
+             py::return_value_policy::reference_internal,
+             "获取指定名称的权重张量")
+        .def("weights", 
+             &axono::core::Module::weights, 
+             py::return_value_policy::reference_internal, 
+             "返回模块中所有权重的映射");
+}

pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "axono"
-version = "0.1.0"
+version = "0.2.0"
 description = "Deep learning framework with tensor operations and neural network modules"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -39,11 +39,7 @@ package-dir = { "" = "python" }
 packages = [
     "axono",
     "axono.core",
-    "axono.data",
-    "axono.models",
     "axono.nn",
-    "axono.train",
-    "axono.viz",
     "axono.core.operators",
     "axono.core.ops"
 ]

python/CMakeLists.txt

@@ -16,26 +16,26 @@ endif()
 
 list(FILTER CORE_SOURCES EXCLUDE REGEX "\\.ipynb_checkpoints/")
 
-pybind11_add_module(axonolib
+pybind11_add_module(libaxono
     src/pybind11_module.cpp
     ${CORE_SOURCES}
 )
 
 if(WITH_CUDA)
-target_compile_definitions(axonolib PRIVATE COMPILED_WITH_CUDA)
+target_compile_definitions(libaxono PRIVATE COMPILED_WITH_CUDA)
 endif()
 
-target_include_directories(axonolib PRIVATE 
+target_include_directories(libaxono PRIVATE 
     ${CMAKE_SOURCE_DIR}/include
 )
 
-target_include_directories(axonolib PRIVATE
+target_include_directories(libaxono PRIVATE
     ${CMAKE_CURRENT_SOURCE_DIR}/include
 )
 
 # Set output properties
-set_target_properties(axonolib PROPERTIES
-    OUTPUT_NAME "axonolib"
+set_target_properties(libaxono PROPERTIES
+    OUTPUT_NAME "libaxono"
     LIBRARY_OUTPUT_DIRECTORY "${CMAKE_SOURCE_DIR}/python/axono/library"
     RUNTIME_OUTPUT_DIRECTORY "${CMAKE_SOURCE_DIR}/python/axono/library"
     PREFIX ""

python/axono/__init__.py

@@ -15,7 +15,7 @@
 
 from .core import DataType, Status, Tensor, operators
 
-__version__ = "0.1.0"
+__version__ = "0.2.0"
 __author__ = "ByteRainLab"
 __description__ = "High performance computing library for big data processing"
 

python/axono/core/__init__.py

@@ -15,7 +15,7 @@
 library_path = os.path.dirname(os.path.dirname(__file__)) + "/library/"
 sys.path.append(library_path)
 
-from axonolib import DataType, Status  # noqa: E402
+from libaxono import DataType, Status  # noqa: E402
 
 from . import operators  # noqa: E402
 from .tensor import Tensor  # noqa: E402

python/axono/core/operators/add.py

@@ -13,7 +13,7 @@
 Axono Add
 """
 
-from axonolib import add as _add
+from libaxono import add as _add
 
 from ..tensor import Tensor
 

python/axono/core/operators/matmul.py

@@ -13,7 +13,7 @@
 Axono Matmul
 """
 
-from axonolib import matmul as _matmul
+from libaxono import matmul as _matmul
 
 from ..tensor import Tensor
 

python/axono/core/ops/relu.py

@@ -13,8 +13,8 @@
 core.ops.Relu()
 """
 
-from axonolib import relu as relu_op
-from axonolib import relu_ as relu_op_
+from libaxono import relu as relu_op
+from libaxono import relu_ as relu_op_
 
 from ..tensor import Tensor
 

python/axono/core/tensor.py

@@ -17,8 +17,8 @@
 import os
 
 import numpy as np
-from axonolib import DataType, Status
-from axonolib import Tensor as _Tensor
+from libaxono import DataType, Status
+from libaxono import Tensor as _Tensor
 
 default_device = os.getenv("axono_default_device", "cpu")