torch.func.linearize ==================== .. currentmodule:: torch.func .. autofunction:: linearize