SetOriginalShape

Description

Sets the OriginShape attribute of an aclTensor.

OriginShape indicates the original shape information of the aclTensor before it passes through the transdata node (if any), that is, the mathematical description of the tensor shape.

Prototype

void SetOriginalShape(const op::Shape &shape)

Parameters

For details about gert::Shape, see gert::Shape in Basic Data Structures and APIs.

Returns

None.

Constraints

None.

Example

1
2
3
4
5
6
7
8

// Set OriginShape of the input to [1, 2, 3, 4, 5].
void Func(aclTensor *input) {
    gert::Shape newShape;
    for (int64_t i = 1; i <= 5; i++) {
        newShape.AppendDim(i);
    }
    input->SetOriginalShape(newShape);
}