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); }
父主题: common_types