ShapeOf

ShapeOf  // Operation that returns the shape of its input tensor

Description

Warning

This op is experimental and subject to change without notice.

Returns the shape of its input argument as a tensor of element type u64.

Inputs

Name Element Type Shape
arg Any Any

Outputs

Name Element Type Shape
output element::u64 {r} where r is the rank of arg’s shape.

Mathematical Definition

\[\mathtt{output} = \mathtt{shapeof}(\mathtt{arg})\]

C++ Interface

class ShapeOf : public ngraph::op::Op

Operation that returns the shape of its input argument as a tensor.

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

ShapeOf(const Output<Node> &arg)

Constructs a shape-of operation.

void validate_and_infer_types()

Throws if the node is invalid.