目录

GetShapeSize

Function Description

Obtains the shape size of the current tensor, that is, the number of elements in the tensor.

Function Prototype

int64_t GetShapeSize() const

Parameters

None

Return Value

Size of the shape during execution

Restrictions

None

Calling Example

Tensor tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}}, // shape
              {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, // format
              kOnDeviceHbm, // placement
              ge::DT_FLOAT16, // dt
              nullptr};
auto shape_size = tensor.GetShapeSize(); // 16*3*224*224