下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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。

参考资源

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);
    }
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词