GetShape接口
接口功能
获取张量对象shape。
接口格式
const std::vector<int64_t> &GetShape() const
接口参数
无。
使用样例
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