torch.cuda.is_current_stream_capturing

torch.cuda.is_current_stream_capturing() [source]

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

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