昇腾社区首页
中文
注册
开发者
下载

c10_npu::NPUStreamGuard::current_device

产品支持情况

[object Object]undefined

功能说明

返回使用此设备guard设置的最新NPU设备。

定义文件

torch_npu\csrc\core\npu\NPUGuard.h

函数原型

[object Object]

参数说明

返回值说明

[object Object]

表示当前设备。

约束说明