2020年10月1日
PyTorch grad_fn的作用以及RepeatBackward, SliceBackward示例
摘要： 变量.grad_fn表明该变量是怎么来的，用于指导反向传播。例如loss = a+b，则loss.gard_fn为<AddBackward0 at 0x7f2c90393748>，表明loss是由相加得来的，这个grad_fn可指导怎么求a和b的导数。 程序示例： import torch w1 = 阅读全文
posted @ 2020-10-01 10:40 Picassooo 阅读(15892) 评论(3) 推荐(2) 编辑