pytorch relu代码

以下是PyTorch中实现ReLU激活函数的代码：

import torch.nn as nn

class ReLU(nn.Module):
    def __init__(self, inplace=False):
        super(ReLU, self).__init__()
        self.inplace = inplace

    def forward(self, x):
        return nn.functional.relu(x, inplace=self.inplace)

可以直接使用`nn.ReLU()`来调用PyTorch自带的ReLU函数，如下所示：

import torch.nn as nn

relu = nn.ReLU()
output = relu(input)