GetTypeByteSize接口

接口功能

获取对应InferDataType类型的字节数。

接口格式

static size_t GetTypeByteSize(InferDataType dataType)

接口参数

使用样例

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);
size_t size = inputsTensor->GetTypeByteSize();

返回值

size_t。