Pytorch下实现自定义激活函数

（每个人错误情况可能不一样，仅供参考。以复现一篇含自定义激活函数论文为例。）
1：拟设计激活函数：

2：代码编辑：

class log_act(nn.Module):
    def __init__(self, alpha, beta, positive_flag=True):
        super(log_act, self).__init__()
        self.positive_flag = positive_flag
        self.a = alpha
        self.C1 = beta
    def forward(self, x):
        x = x.detach().numpy()
        C0 = np.exp(