torch.cuda.is_current_stream_capturing

torch.cuda.is_current_stream_capturing()

如果當前 CUDA 流上正在進行 CUDA 計算圖捕獲，則返回 True，否則返回 False。

如果當前裝置上不存在 CUDA 上下文，則返回 False，不初始化上下文。