GraphNode
设置Operation
- 功能
设置节点operation。
- 原型
GraphNode.SetOperation (op: operation) -> torch_atb.GraphNode
- 参数
atb::operation - 注意
调用接口失败，会抛出异常终止运行。
获取节点outTensor
- 功能
获取节点outTensor。
- 原型
GraphNode.GetOutput (index: size_t) -> torch_atb.GraphNode
- 参数
index
- 返回值
string
- 注意
调用接口失败，会抛出异常终止运行。
- 使用示例
1 2 3
# layer1为添加到计算图中的节点 layer1 = builder.add_node([x, y], elewise_add) add_out = layer1.get_output(0)
设置节点流id
- 功能
设置节点流id。
- 原型
GraphNode.SetStreamId (streamId: uint32_t) -> torch_atb.GraphNode
- 参数
streamId
- 注意
调用接口失败，会抛出异常终止运行。
- 使用示例
1 2
# layer1为添加到计算图中的节点 layer1.set_stream_id(1)
获取节点流id
- 功能
获取节点流id。
- 原型
GraphNode.GetStreamId () -> torch_atb.GraphNode
- 返回值
streamId
- 注意
调用接口失败，会抛出异常终止运行。
- 使用示例
1 2
# layer1为添加到计算图中的节点 id = layer1.get_stream_id(1)
父主题： 组图接口