大家好,我是爱编程的喵喵。双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中。从事机器学习以及相关的前后端开发工作。曾在阿里云、科大讯飞、CCF等比赛获得多次Top名次。现为优快云博客专家、人工智能领域优质创作者。喜欢通过博客创作的方式对所学的知识进行总结与归纳,不仅形成深入且独到的理解,而且能够帮助新手快速入门。
本文主要介绍了leaf variable & with torch.no_grad & -=,希望能对大家有所帮助。
文章目录
本文以下列三层全连接神经网络代码为例讲解(包括输入层,中间层<含一个linear和一个reluctant>,输出层)。该代码没有采用torch.nn这个库,但使用了pytorch的自动求到,因此需要提前初始化参数W1和W2,手动写出y_pred和损失函数的表达式、手动更新参数。
# 定义样本数,输入层维度,隐藏层维度,输出层维度
n, d_input, d_hidden, d_output = 64, 1000, 100, 10
# 随机创建一些训练数据
X
订阅专栏 解锁全文
413

被折叠的 条评论
为什么被折叠?



