(ctx, grad)
| 63 | |
| 64 | @staticmethod |
| 65 | def backward(ctx, grad): |
| 66 | u, v = ctx.saved_tensors |
| 67 | d_u = triangular_toeplitz_multiply_(grad.flip(-1), v).flip(-1) |
| 68 | d_v = triangular_toeplitz_multiply_(grad.flip(-1), u).flip(-1) |
| 69 | return d_u, d_v |
| 70 | |
| 71 | class TriangularToeplitzMultFast(torch.autograd.Function): |
| 72 | @staticmethod |
no test coverage detected