PyTorch Autograd Function

class MyFunction(torch.autograd.Function):
  """Some Information about MyFunction"""

  @staticmethod
  def forward(ctx, input):

    return

  @staticmethod
  def backward(ctx, grad_output)

    return