torch.cuda.is_current_stream_capturing

torch.cuda.is_current_stream_capturing() [source]

Возвращает True, если захват графика CUDA выполняется на текущем потоке CUDA, в противном случае — False.

Если контекст CUDA не существует на текущем устройстве, возвращает False без инициализации контекста.