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]的数据使用。
参数名 |
输入/输出 |
说明 |
---|---|---|
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表示成功,返回其他值表示失败。
父主题: 基础接口