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

const std::string &description() const

Get the string name for the type of the node, such as Add or Multiply. The class name, must not contain spaces as it is used for codegen.

Return

A const reference to the node’s type name

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.

Output<Node> get_as_output() const

Return the equilent Output<Node>

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.