mindietorch.npu.current_stream
函数功能
获取device的当前stream，当device为None，返回当前device的当前stream。
函数原型
def current_stream(input_device=None)
约束说明
mindietorch.npu.current_stream接口由于存在参数校验，在输入非法数据时，可能会抛出异常。故用户必须在try/except语句块内进行调用以及异常处理，防止在使用的过程中出现异常抛出导致程序退出的情况。
参数说明
参数名称 |
参数类型 |
输入/输出 |
参数说明 |
|---|---|---|---|
input_device |
int or str or torch.device |
输入 |
NPU的device。 input_device需要为合法的NPU device。 |
返回值说明
mindietorch.npu.Stream。
父主题： 函数方法