def __init__(self, stream: Optional[torch_aie.npu.Stream])
函数功能
初始化函数。
函数原型
def __init__(self, stream: Optional[torch_aie.npu.Stream])
参数说明
参数名 |
参数类型 |
描述 |
|---|---|---|
stream |
mindietorch.npu.Stream |
必选；MindIE-Torch的Stream。 具体描述请参见class mindietorch.npu.Stream。 |
返回值说明
无