ReLU¶
- class torch.nn.ReLU(inplace=False)[source][source]¶
应用逐元素 ReLU 线性单元函数。
- 参数:
inplace(布尔值)- 可选地进行原地操作。默认:
False
- 形状:
输入: ,其中 表示任意数量的维度。
输出: ,与输入具有相同的形状。
示例:
>>> m = nn.ReLU() >>> input = torch.randn(2) >>> output = m(input) An implementation of CReLU - https://arxiv.org/abs/1603.05201 >>> m = nn.ReLU() >>> input = torch.randn(2).unsqueeze(0) >>> output = torch.cat((m(input), m(-input)))