run_graph_with_stream_async
功能说明
异步运行图。
函数原型
1 | run_graph_with_stream_async(graph_id: int, stream: int, inputs: List[Tensor]) -> List[Tensor] |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
graph_id |
输入 |
图ID。 |
stream |
输入 |
指定图在哪个Stream上运行。 |
inputs |
输入 |
输入张量列表,为Device上的内存空间。 |
返回值说明
(List[Tensor])输出张量列表,为Device上的内存空间。
父主题: Session接口