GetOutputShape

Description

Obtains the pointer to an output shape based on the operator output index. The output index refers to the actual index after operator instantiation, not the index in the prototype definition.

Prototype

Shape *GetOutputShape(const size_t index)

Parameters

Returns

Pointer to a specified output shape. If the input index is invalid, a null pointer is returned.

For details about the definition of the shape type, see Shape.

Restrictions

None

Example

ge::graphStatus InferShapeForReshape(InferShapeContext *context) {
  const gert::Shape *x_shape = context->GetInputShape(0);        // Obtain the shape of input 0.
  const gert::Tensor *shape_tensor = context->GetInputTensor(1); // Obtain the data dependency of input tensor 1.
  gert::Shape *output_shape = context->GetOutputShape(0);
  if (x_shape == nullptr || shape_tensor == nullptr || output_shape == nullptr) {
    // Defensive programming. In an undesirable scenario, an error is printed and a failure message is returned.
    return ge::GRAPH_FAILED;
  }
  // ...
}