设置多线程启用¶ class torch.autograd.grad_mode.set_multithreading_enabled(mode)[源代码]¶ 上下文管理器,用于设置多线程向后兼容性为开启或关闭。 set_multithreading_enabled 将根据其参数 mode 启用或禁用多线程反向传播。 它可以作为上下文管理器或函数使用。 此上下文管理器是线程本地的;它不会影响其他线程中的计算。 Parameters mode (bool) – 标志是否启用多线程反向传播(True),或禁用(False)。 注意 此API不适用于前向模式自动微分。 clone()[源代码]¶ 创建此类的副本 Return type 设置多线程启用