評價此頁

torch.cuda.is_current_stream_capturing#

torch.cuda.is_current_stream_capturing()[原始碼]#

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

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

返回型別

布林值