下载
中文
注册

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提供,具体参见《应用开发接口》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]的数据使用。

参数名

输入/输出

说明

tensor

输入

待初始化参数的tensor,通过aclCreateTensor创建。

viewDims

输入

tensor的ViewShape维度值。

viewDimsNum

输入

tensor的ViewShape维度数。

dataType

输入

tensor的数据类型。

stride

输入

tensor的view的各个维度步长,取值不支持负数。

offset

输入

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

format

输入

tensor的数据排布格式。

storageDims

输入

tensor的StorageShape维度值。

storageDimsNum

输入

tensor的StorageShape维度数。

tensorDataAddr

输入

tensor在Device侧的存储地址。

返回值说明

返回0表示成功,返回其他值表示失败。