set_multithreading_enabled
class torch.autograd.set_multithreading_enabled(mode)
Context-manager that sets multithreaded backwards on or off.
"set_multithreading_enabled" will 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.
Parameters:
      mode (bool) -- Flag whether to enable multithreaded
      backwards ("True"), or disable ("False").
Note:
 This API does not apply to forward-mode AD.
