下载
中文
注册

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提供,具体参见aclDataTypeaclFormat
  • 关于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]的数据使用。

参数名

输入/输出

说明

viewDims

输入

tensor的ViewShape维度值。

viewDimsNum

输入

tensor的ViewShape维度数。

dataType

输入

tensor的数据类型。

stride

输入

tensor的view的各个维度步长。

offset

输入

tensor首元素相对于storage的偏移。

format

输入

tensor的数据排布格式。

storageDims

输入

tensor的StorageShape维度值。

storageDimsNum

输入

tensor的StorageShape维度数。

tensorData

输入

tensor在Device侧的存储地址。

返回值说明

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