TensorForge

TensorForge 是一个从零实现的轻量级深度学习框架，用于学习和理解现代深度学习框架的核心原理。

特性

• Tensor 张量运算
• 自动微分
• 神经网络层
• 损失函数
• 优化器
• CPU / CUDA 支持（规划中）

示例

import tensorforge as tf

x = tf.Tensor([1.0, 2.0, 3.0], requires_grad=True)
y = x * x

loss = y.sum()
loss.backward()

print(x.grad)

开发计划

• Tensor 基础运算
• Autograd 自动微分
• 神经网络模块
• 损失函数与优化器
• CUDA 后端
• 模型保存与加载

项目目标

通过从零实现 TensorForge，深入理解 Tensor、计算图、自动微分以及神经网络训练等深度学习框架核心机制。

License

MIT License