torch.cuda.is_current_stream_capturing

torch.cuda.is_current_stream_capturing() [исходный код]

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

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

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

bool