GetOutputShape

Function 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.

Function Prototype

const StorageShape *GetOutputShape(size_t index) const;

Parameters

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

Return Value

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

For details about the definition of the StorageShape type, see StorageShape.

Restrictions

None

Calling Example

ge::graphStatus Tiling4ConcatD(TilingContext* context) {
  auto out_shape = context->GetOutputShape(0);
  // ...
}