花費 31 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