设置多线程启用

class torch.autograd.grad_mode.set_multithreading_enabled(mode)[源代码]

一个上下文管理器，用于控制多线程反向传播的开启或关闭。

set_multithreading_enabled 根据参数 mode 来启用或禁用多线程反向传播。它既可以作为上下文管理器使用，也可以直接作为函数调用。

这个上下文管理器是线程局部的，不会影响其他线程中的计算。

参数

mode (bool) – 是否启用多线程反向传播的标志（True），或禁用（False）。

注意

此 API 不适用于 前向模式自动微分。

clone()[源代码]

创建这个类的副本

返回类型

set_multithreading_enabled