ConvertToTensor

功能说明

将不同数据类型的host侧数据，转为一个host侧aclTensor对象。

函数原型

aclTensor *ConvertToTensor(const aclIntArray *value, DataType dataType)

aclTensor *ConvertToTensor(const aclBoolArray *value, DataType dataType)

aclTensor *ConvertToTensor(const aclFloatArray *value, DataType dataType)

aclTensor *ConvertToTensor(const aclFp16Array *value, DataType dataType)

aclTensor *ConvertToTensor(const aclBf16Array *value, DataType dataType)

aclTensor *ConvertToTensor(const aclScalar *value, DataType dataType)

template<typename T>

aclTensor *ConvertToTensor(const T *value, uint64_t size, DataType dataType)

参数说明

返回值说明

返回转换后的host侧tensor。

约束说明

入参指针不能为空。

调用示例

1
2
3
4
5
6
7

// 分别将一个aclScalar和int64_t转为host侧tensor
void Func(aclOpExecutor *executor) {
    int64_t val = 5;
    aclScalar *scalar = executor->AllocScalar(val);
    aclTensor *tensor = executor->ConvertToTensor(scalar, DT_INT64);
    tensor = executor->ConvertToTensor(&val, 1, DT_INT64);
}