SetData
函数功能
针对通过AllocHostTensor申请得到的host侧tensor,设置指定位置的数据。
函数原型
参数说明
- 设置指定索引处的值的接口说明
参数
输入/输出
说明
index
输入
需要修改aclTensor的第几个元素。
value
输入
将aclTensor的指定元素修改为value的值。
dataType
输入
数据类型为op::DataType(即ge::DataType)。将value转为指定的dataType后,再写入aclTensor。
- 用一块已有内存初始化tensor数据的接口说明
参数
输入/输出
说明
value
输入
指向需要写入aclTensor的数据内存指针。
size
输入
需要写入的元素个数。
dataType
输入
数据类型为op::DataType(即ge::DataType)。将数据转为指定的dataType后,再写入aclTensor。
返回值说明
无
约束说明
入参指针不能为空。
调用示例
1 2 3 4 5 6 |
// 初始化一块int64_t内存,分别将input的前10个数字,置为该内存的内容。并将input的第11个数字置为myArray的第一个数字。 void Func(const aclTensor *input) { int64_t myArray[10]; input->SetData(myArray, 10, DT_INT64); input->SetData(10, myArray[0], DT_INT64); } |
父主题: common_types