GetOutputElement

GetOutputElement  // Operation to select a unique output from an op

Description

Accesses an output of a node.

Inputs

Name Element Type Shape
arg Any Any

Attributes

Name Description
n The output number from the node arg

Outputs

Name Element Type Shape
output Depends on arg Depends on arg

C++ Interface

class GetOutputElement : public ngraph::op::Op

Operation to get an output from a node.

Public Functions

GetOutputElement(const std::shared_ptr<Node> &arg, size_t n)

Constructs a get-tuple-element operation.

Parameters
  • arg: The input tuple.
  • n: The index of the tuple element to get.

void validate_and_infer_types()

Throws if the node is invalid.

size_t get_n() const

Return
The index of the tuple element to get.