@staticmethod
def forward(ctx, input):
    ctx.save_for_backward(input)
    return np.maximum(input, 0)
@staticmethod
def forward(ctx, input):
    ctx.save_for_backward(input)
    return np.maximum(input, 0)