@staticmethod
def forward(ctx, input, weight):
    ctx.save_for_backward(input, weight)
    return (input @ weight)
@staticmethod
def forward(ctx, input, weight):
    ctx.save_for_backward(input, weight)
    return (input @ weight)