GetDimSizes

Description

Obtains the shape of a tensor. The size of each dimension is represented by an std::vector<int64_t>.

Prototype

std::vector<int64_t> GetDimSizes() const

Parameters

None

Returns

An std::vector<int64_t> for the tensor shape.

Restrictions

None

Example

std::vector<int64_t> dims = shape->GetDimSizes();