ngraph.opset6.shape_of

ngraph.opset6.shape_of(data: Union[_pyngraph.Node, int, float, numpy.ndarray], output_type: str = 'i64', name: Optional[str] = None) _pyngraph.Node

Return a node which produces a tensor containing the shape of its input data.

Parameters
  • data – The tensor containing the input data.

  • output_type – Output element type.

Returns

ShapeOf node