tabensemb.utils.utils.torch\_with\_grad ======================================= .. currentmodule:: tabensemb.utils.utils .. autoclass:: torch_with_grad :show-inheritance: .. rubric:: Methods .. automethod:: __init__ .. autosummary:: .. HACK -- the point here is that we don't want this to appear in the output, but the autosummary should still generate the pages. .. autosummary:: :toctree: torch_with_grad.__call__ torch_with_grad._wrap_generator torch_with_grad.clone