随笔分类 -  python

requires_grad，grad_fn，grad的含义及使用
摘要：requires_grad: 如果需要为张量计算梯度，则为True，否则为False。我们使用pytorch创建tensor时，可以指定requires_grad为True（默认为False）， grad_fn： grad_fn用来记录变量是怎么来的，方便计算梯度，y = x*3,grad_fn记录 阅读全文
posted @ 2021-01-18 16:36 zpcode 阅读(137) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示