2020年5月28日

pytorch .detach() .detach_() 和 .data用于切断反向传播(转载)

摘要: pytorch中 .detach() .detach_() 和 .data均能用于切断反向传播,取出本体的tensor数据,舍弃了grad,grad_fn等额外反向图计算过程需保存的额外信息。 .data和.detach()取出本体tensor后仍与原数据共享内存,在使用in-place操作后,会修 阅读全文

posted @ 2020-05-28 10:06 那抹阳光1994 阅读(533) 评论(0) 推荐(0) 编辑

导航