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

SetIntData

函数功能

针对通过AllocHostTensor申请得到的host侧tensor,用一块int64_t类型的内存初始化tensor数据。

函数原型

void SetIntData(const int64_t *value, uint64_t size, op::DataType dataType)

参数说明

参数

输入/输出

说明

value

输入

指向需要写入aclTensor的数据内存指针。

size

输入

需要写入的元素个数。

dataType

输入

数据类型为op::DataType(即ge::DataType),将数据转为指定的dataType后,再写入aclTensor。

返回值说明

约束说明

入参指针不能为空。

调用示例

// 初始化一块int64内存,赋值给input的前10个元素
void Func(const aclTensor *input) {
    int64_t myArray[10];
    input->SetData(myArray, 10, DT_INT64);
}
搜索结果
找到“0”个结果

当前产品无相关内容

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