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]的数据使用。
参数名 |
输入/输出 |
说明 |
|---|---|---|
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。
参考资源
aclTensor的定义与torch.Tensor相似，由一块连续或非连续的内存地址和一系列描述信息（如stride、offset等）组成。Tensor根据shape、stride、offset信息，可自由取出内存中的数据，也可获得非连续的内存（如图1中y）。
父主题： 基础接口
