- class torch.autograd.set_multithreading_enabled(mode)[source]¶
Context-manager that sets multithreaded backwards on or off.
set_multithreading_enabledwill enable or disable multithreaded backwards based on its argument
mode. It can be used as a context-manager or as a function.
This context manager is thread local; it will not affect computation in other threads.
mode (bool) – Flag whether to enable multithreaded backwards (
True), or disable (
This API does not apply to forward-mode AD.