class MyFunction(torch.autograd.Function):
"""Some Information about MyFunction"""
@staticmethod
def forward(ctx, input):
return
@staticmethod
def backward(ctx, grad_output)
return
class MyFunction(torch.autograd.Function):
"""Some Information about MyFunction"""
@staticmethod
def forward(ctx, input):
return
@staticmethod
def backward(ctx, grad_output)
return