其他分享
首页 > 其他分享> > torch.nn.ReLU(inplace=True)的具体含义:

torch.nn.ReLU(inplace=True)的具体含义:

作者:互联网

首先根据源文档中的ReLU(x)=max(0,x),得出结论。大于0的数值不变,小于0的数据变成0。

这是对应的文档链接:https://pytorch.org/docs/1.2.0/nn.html#torch.nn.ReLU  Ps:需要自取。

参数:inplace为True,将会改变输入的数据 ,否则不会改变原输入,只会产生新的输出。

好处:省去了反复申请与释放内存的时间,直接代替原来的值。

测试代码:

import torch
from torch import nn
m = nn.ReLU()
# 随机生成5个数 有正有负。
input = torch.randn(5)
# 打印 随机生成的数
print(input)
output = m(input)
# 经过nn.ReLU()作用之后的数
print(output)
# 结果
# tensor([-0.7706, -0.1823,  0.2687,  0.2796, -1.7201])
# tensor([0.0000,   0.0000,  0.2687,  0.2796,  0.0000])

任务完成,下班。

 

标签:nn,torch,inplace,ReLU,import,0.0000,input
来源: https://www.cnblogs.com/kuangmeng/p/15598911.html