花费 5 ms
PyTorch grad_fn的作用以及RepeatBackward, SliceBackward示例

变量.grad_fn表明该变量是怎么来的,用于指导反向传播。例如loss = a+b,则loss.gard_fn为<AddBackward0 at 0x7f2c90393748>,表明loss是由相加得来的,这个grad_fn可指导怎么求a和b的导数。 程序示例 ...

Thu Oct 01 18:40:00 CST 2020 3 7255

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM