DelTensorByName接口
接口功能
根据Name删除Tensor。
接口格式
Status DelTensorByName(const std::string &name)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
name |
是 |
tensor名字。 |
合法的字符串类型。 |
使用样例
mindie_llm::InferRequestId runtimeReqId(0);
std::shared_ptr<mindie_llm::InferRequest> runtimeRequest(runtimeReqId);
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);
runtimeRequest->AddTensor(name, inputsTensor);
TensorPtr tensor = std::make_shared<mindie_llm::InferTensor>();
Status ret = request->DelTensorByName("INPUT_IDS");
返回值
Status对象。
父主题: InferRequest