mindietorch.npu.stream
函数功能
与Python的with配套使用，用于stream的切换。
函数原型
def stream(input_stream: Optional["mindietorch.npu.Stream"]) -> StreamContext
约束说明
mindietorch.npu.stream接口由于存在参数校验，在输入非法数据时，可能会抛出异常。故用户必须在try/except语句块内进行调用以及异常处理，防止在使用的过程中出现异常抛出导致程序退出的情况。
参数说明
参数名称 |
参数类型 |
输入/输出 |
参数说明 |
|---|---|---|---|
input_stream |
mindietorch.npu.Stream |
输入 |
与with配套使用，将with结构体中的stream切换为input_stream。 input_stream需要为合法的stream。 |
返回值说明
返回一个mindietorch.npu.StreamContextl类型的上下文管理器。
父主题： 函数方法