GetInputShape

Function Usage

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

Prototype

const Shape *GetInputShape(const size_t index) const

Parameters

Parameter	Input/Output	Description
index	Input	Operator input index, starting from 0.

Returns

Input shape pointer. If the index is invalid, a null pointer is returned.

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

Constraints

None

Examples

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;
  }
  // ...
}

Parent topic: InferShapeContext