r""" Override this function for custom backward calculation.
(ctx, *out_grads)
| 77 | |
| 78 | @staticmethod |
| 79 | def backward(ctx, *out_grads): |
| 80 | r""" |
| 81 | Override this function for custom backward calculation. |
| 82 | """ |
| 83 | raise NotImplementedError( |
| 84 | "You must implement the backward function for custom autograd.Function." |
| 85 | ) |