设置aclTensor的数据类型。
void SetDataType(op::DataType dataType)
|
参数 |
输入/输出 |
说明 |
|---|---|---|
|
dataType |
输入 |
数据类型为op::DataType(即ge::DataType),将aclTensor设置为dataType类型。
说明:
ge::DataType介绍参见《基础数据结构和接口参考》中“ge::DataType”。 |
无
无
1 2 3 4 |
// 将input的数据类型设置为int64 void Func(const aclTensor *input) { input->SetDataType(DT_INT64); } |