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