set_multithreading_enabled

class torch.autograd.grad_mode.set_multithreading_enabled(mode) [исходный код]

Менеджер контекста, который включает или отключает многопоточную обратную передачу.

Обычно, когда используются устройства ускорителей, обратный проход выполняется в рабочих потоках, специфичных для устройства. Движок создает эти потоки в зависимости от количества доступных устройств и повторно использует их в последующих итерациях.

Когда mode=False, обратный проход выполняется в вызывающем потоке. mode=True восстанавливает поведение по умолчанию.

Эту конструкцию можно использовать как менеджер контекста или как функцию. Она локальна для потока и не влияет на вычисления в других потоках.

Параметры:

mode (bool) – Включать ли многопоточную обратную передачу (True, по умолчанию) или отключать ее (False).

Примечание

Этот API не применяется к прямому автоматическому дифференцированию, которое никогда не использует многопоточность.

clone() [исходный код]

Создает копию этого класса

Тип возвращаемого значения:

set_multithreading_enabled