SetData

函数功能

针对通过AllocHostTensor申请得到的host侧tensor，设置指定位置的数据。

函数原型

• 针对tensor，设置指定索引处的值

  void SetData(int64_t index, const T value, op::DataType dataType)

• 针对tensor，用一块已有内存初始化tensor数据

  void SetData(const T *value, uint64_t size, op::DataType dataType)

参数说明

返回值说明

无

约束说明

入参指针不能为空。

调用示例

// 初始化一块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);
}