项目简介
Tensor库是一个基于C++和Python的深度学习库,它提供了类似于NumPy和PyTorch的接口,用于处理多维数组和构建深度神经网络。该库轻量且依赖较少,适合在嵌入式设备和移动平台上使用。
项目的主要特性和功能
- 多维数组操作:提供类和工具处理多维数组,支持BLAS/LAPACK库,同时有自有的朴素实现作为后备。
- 深度神经网络:支持多种神经网络层(如FeedForward、Conv2D、RNN、LSTM、Pooling、Dropout、BatchNormalization2D)和优化器(如SGD、Adagrad、RMSProp、Adam)。
- 模型保存与恢复:采用protobuf格式保存和恢复模型。
- Python接口:提供Python包装器,支持大部分C++功能,还包含实验性的自动微分模块。
安装使用步骤
C++ 部分
- 构建库:
bash cd tensor mkdir build && cd build cmake .. make
- 集成到项目:暂未给出具体集成步骤,需根据实际情况操作。
Python 部分
- 安装库:暂未给出安装命令,需等待补充。
- 使用示例: ```python import tensor as ts from tensor.nn import Conv2D, Linear
conv = Conv2D(32, 16, kernel_size=3, stride=1, padding=1) linear = Linear(16, 10) model = ts.nn.Sequential([conv, linear]) ```
更多详细的使用说明和API文档,请参考Tensor库文档。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】