您好,欢迎来到叨叨游戏网。
搜索
您的当前位置:首页PyTorch中ReLU的inplace

PyTorch中ReLU的inplace

来源:叨叨游戏网

0 - inplace

  在pytorch中,nn.ReLU(inplace=True)和nn.LeakyReLU(inplace=True)中存在inplace字段。该参数的inplace=True的意思是进行原地操作,例如:

  • x=x+5是对x的原地操作
  • y=x+5,x=y不是对x的原地操作

  所以,如果指定inplace=True,则对于上层网络传递下来的tensor直接进行修改,可以少存储变量y,节省运算内存。

inplace=True means that it will modify the input directly, without allocating 
any additional output. It can sometimes slightly decrease the memory usage, 
but may not always be a valid operation (because the original input is destroyed). 
However, if you don’t see an error, it means that your use case is valid.

  如果你使用了in-place operation而没有报错的话,那么你可以确定你的梯度计算是正确的。

1 - 参考资料

转载于:https://www.cnblogs.com/CZiFan/p/10790765.html

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- gamedaodao.net 版权所有 湘ICP备2024080961号-6

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务