aclCreateTensor

函数功能

创建aclTensor。

函数原型

aclTensor *aclCreateTensor(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 *tensorData)

参数说明

• 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]的数据使用。

返回值说明

成功则返回创建好的aclTensor，否则返回nullptr。

参考资源

aclTensor的定义与torch.Tensor相似，由一块连续或非连续的内存地址和一系列描述信息（如stride、offset等）组成。Tensor根据shape、stride、offset信息，可自由取出内存中的数据，也可获得非连续的内存（如图1中y）。

图1 tensor逻辑结构

调用示例

• 以图1为例，x tensor创建过程如下：

  aclTensor *CteateXTensor()
  {
      int64_t viewDims = {2, 4};       
      int64_t stride = {4, 1};          // 第1维步长4，第2维步长1
      int64_t storageDims = {2, 4};    
      return aclCreateTensor(viewDims, 2, ACL_FLOAT16, stride, 0, ACL_FORMAT_ND, storageDims, 2, nullptr);
  }

• 以图1为例，x对应的转置x^T tensor创建过程如下：

  aclTensor *CreateXTransposedTensor()
  {
      int64_t viewDims = {4, 2};       
      int64_t stride = {1, 4};         // 转置跨度通过stride表示
      int64_t storageDims = {2, 4};    
      return aclCreateTensor(viewDims, 2, ACL_FLOAT16, stride, 0, ACL_FORMAT_ND, storageDims, 2, nullptr);
  }