获取索引的数据地址以及可供使用的数据大小。
原型设计
std::pair<std::uint64_t, std::size_t> data(const std::vector<std::int32_t> idx = {});参数
下表列出了data 函数实参。
| 类型 | 名称 | 描述 |
|---|---|---|
| const std::vector<std::int32_t> | idx | 要访问的数据的索引，其维度与张量形状相同 |
返回
一对索引数据地址以及可供使用的数据字节大小（以字节为单位）。用法
vart::TensorBuffer* tb;
std::tie(data_addr, tensor_size) = tb->data({0,0,0,0});