GetShape接口

接口功能

获取张量对象shape。

接口格式

const std::vector<int64_t> &GetShape() const

接口参数

无。

使用样例

构造一个名为"INPUT_IDS"，数据类型为INT64，数据维度为{1，2}的Tensor对象，然后获取shape信息。

1
2
3
4
5
std::string name = "INPUT_IDS";
mindie_llm::InferDataType dataType = mindie_llm::InferDataType::TYPE_INT64;
std::vector<int64_t> dataShape = {1, 2};
auto inputsTensor = std::make_shared<mindie_llm::InferTensor>(name, dataType, dataShape);
std::vector<int64_t> shape = inputsTensor->GetShape();

返回值

tensor的数据维度，类型为vector<int64_t>。

父主题： InferTensor