tabensemb.utils.utils.torch_with_grad._wrap_generator

method

torch_with_grad._wrap_generator(func)

Wrap each generator invocation with the context manager