aclInitTensor

函数功能

初始化给定tensor的参数。

当用户想复用已创建aclTensor时，可使用该接口先重置aclTensor的各项属性。

函数原型

aclnnStatus aclInitTensor(aclTensor *tensor, const int64_t *viewDims, uint64_t viewDimsNum, aclDataType dataType, const int64_t *stride, int64_t offset, aclFormat format, const int64_t *storageDims, uint64_t storageDimsNum, void *tensorDataAddr)

参数说明

• aclDataType、aclFormat定义由AscendCL提供，具体参见《应用开发接口》中aclDataType、aclFormat。
• 关于aclTensor的StorageShape和ViewShape：
  • ViewShape表示aclTensor的逻辑shape，是Tensor在实际使用时需要用到的大小。
  • StorageShape表示aclTensor的实际物理排布shape，是Tensor在内存上实际存在的大小。

  举例如下：

  • StorageShape为[10, 20]：表示该aclTensor在内存上是按照[10, 20]排布的。
  • ViewShape为[2, 5, 20]：在算子使用时，表示该aclTensor可被视为一块[2, 5, 20]的数据使用。

返回值说明

返回0表示成功，返回其他值表示失败，返回码列表参见返回码。

约束与限制

无

调用示例

关键代码示例如下，不可以直接拷贝运行，仅供参考。

std::vector<int64_t> viewDims = {2, 4};
std::vector<int64_t> stride = {4, 1};
std::vector<int64_t> storageDims = {2, 4};  
// tensor为复用的已经创建的aclTensor
// deviceAddr为tensor在Device侧的存储地址
auto ret = aclInitTensor(tensor, viewDims.data(), viewDims.size(), ACL_FLOAT16, stride.data(), 0, aclFormat::ACL_FORMAT_ND, storageDims.data(), storageDims.size(), deviceAddr);