pytorch中激活函数ReLU

nn.ReLU(inplace=True)

Parameters
inplace – can optionally do the operation in-place. Default: False

ReLU(x)=max(0,x)
在这里插入图片描述
参数inplace=True:
inplace为True,将会改变输入的数据 ,否则不会改变原输入,只会产生新的输出

注: 产生的计算结果不会有影响。利用in-place=True计算可以节省内(显)存,同时还可以省去反复申请和释放内存的时间。但是会对原变量覆盖,只要不带来错误就用。

m = nn.ReLU()
n = nn.ReLU(inplace=True)
input = autograd.Variable(torch.randn(5))
inputTemp = input
print(input)
print(m(input))
print(input,'\n')
print(inputTemp)
print(n(inputTemp))
print(inputTemp)

#运行结果
tensor([ 0.2750, -0.1387, -0.6451, -0.8800,  0.4505])
tensor([0.2750, 0.0000, 0.0000, 0.0000, 0.4505])
tensor([ 0.2750, -0.1387, -0.6451, -0.8800,  0.4505]) 

tensor([ 0.2750, -0.1387, -0.6451, -0.8800,  0.4505])
tensor([0.2750, 0.0000, 0.0000, 0.0000, 0.4505])
tensor([0.2750, 0.0000, 0.0000, 0.0000, 0.4505])

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值