Shortcuts

设置多线程启用

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

上下文管理器,用于设置多线程向后兼容性为开启或关闭。

set_multithreading_enabled 将根据其参数 mode 启用或禁用多线程反向传播。 它可以作为上下文管理器或函数使用。

此上下文管理器是线程本地的;它不会影响其他线程中的计算。

Parameters

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

注意

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

clone()[源代码]

创建此类的副本

Return type

设置多线程启用

优云智算