设置aclTensor的数据类型。
void SetDataType(op::DataType dataType)
参数 |
输入/输出 |
说明 |
---|---|---|
dataType |
输入 |
数据类型为op::DataType(即ge::DataType),将aclTensor设置为dataType类型。 |
无
无
1 2 3 4 |
// 将input的数据类型设置为int64 void Func(const aclTensor *input) { input->SetDataType(DT_INT64); } |