很多地方有inplace这个参数,最近遇到的是nn.ReLU(inplace=True)
inplace为True,将会改变输入的数据 ,否则不会改变原输入,只会产生新的输出
inplace:can optionally do the operation in-place. Default: False
注: 产生的计算结果不会有影响。利用in-place计算可以节省内(显)存,同时还可以省去反复申请和释放内存的时间。但是会对原变量覆盖,只要不带来错误就用。
inplace=True
猜你喜欢
转载自blog.csdn.net/hxxjxw/article/details/120637104
今日推荐
周排行