摘要: 下面是如何自己构建一个层,分为包含自动反向求导和手动反向求导两种方式,后面会分别构建网络,对比一下结果对不对。 关于Pytorch中的结构层级关系。 最为底层的是torch.relu()、torch.tanh()、torch.ge()这些函数,这些函数个人猜测就是直接用Cuda写成的,并且封装成了p 阅读全文
posted @ 2019-02-21 19:15 阿刚的代码进阶之旅 阅读(1464) 评论(0) 推荐(0) 编辑