set_graph_output

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Sets the output of a graph.

set_graph_output(tensor: TensorHolder, output_index: int) -> None

Parameter	Input/Output	Description
tensor	Input	Holder of the tensor to be set as the output.
output_index	Input	Output index.

None

If the parameter type is incorrect, TypeError is thrown.

Parent topic: GraphBuilder