easytorch
使用Python的numpy实现的简易深度学习框架，API与pytorch基本相同，实现了自动求导、基础优化器、layer等。
1 文档目录
2 Quick Start
from easytorch.layer import Linear, Tanh, Sequential
from easytorch.optim import SGD
import easytorch.functional as F
# Create a model, optimizer, loss function
model = Sequential(
Linear(1, 5),
Tanh(),
Linear(5, 1)
)
opt = SGD(model.parameters(), lr=3e-4)
loss_fn = F.mse_loss
# train the mod
2023-05-15 20:47:00
35KB
1
主要介绍了TensorFlow Autodiff自动微分详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2022-02-28 10:30:36
30KB
1