torch.cuda.is_current_stream_capturing

torch.cuda.is_current_stream_capturing()

如果 CUDA 圖形捕獲正在目前的 CUDA stream 上進行中，則回傳 True，否則回傳 False。

如果目前的裝置上不存在 CUDA 環境，則回傳 False 而不初始化環境。