def forward(ctx, input): inverse = torch.inverse(input) ctx.save_for_backward(inverse) return inverse