【Pytorch】Inplace operation(待完善)

什么是Inplace操作

本质上讲,inplace操作是指将新值赋到原变量地址上的操作,目的是节约内存空间。例如inplace操作‘x += 1’,假设原x值为0,储存在id为139973228650216的内存地址中,在进行该inplace操作后,x为1且仍储存在139973228650216中。

哪些是Inplace操作,哪些不是

任何有一个'_'后缀改变张量的操作都是inplace操作。例如x.squeeze_(),x.unsqueeze_()操作将改变x。x.squeeze(),x.unsqueeze()则不会。

Pytorch中 torch.relu()和torch.sigmoid()等激活函数不是inplace操作,其中ReLU可通过设置inplace=True进行inplace操作。

x += res是inplace操作,x = x + res不是(详情见官网论坛Adam Paszke的回答)。

引起modified by an inplace operation的原因

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

注意:x += res是inplace operation,x = x + res不是

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值