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 |
---|---|---|
|
Any |
Any |
Outputs¶
Name |
Element Type |
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
orMultiply
. 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.
-
const std::string &