会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
阿刚的代码进阶之旅
博客园
首页
新随笔
联系
管理
订阅
2019年2月21日
pytorch构建自己设计的层
摘要: 下面是如何自己构建一个层,分为包含自动反向求导和手动反向求导两种方式,后面会分别构建网络,对比一下结果对不对。 关于Pytorch中的结构层级关系。 最为底层的是torch.relu()、torch.tanh()、torch.ge()这些函数,这些函数个人猜测就是直接用Cuda写成的,并且封装成了p
阅读全文
posted @ 2019-02-21 19:15 阿刚的代码进阶之旅
阅读(1464)
评论(0)
推荐(0)
编辑
公告