SetDataType

函数功能

设置Tensor的数据类型。

函数原型

void SetDataType(const ge::DataType data_type)

参数说明

返回值说明

无。

约束说明

无。

调用示例

StorageShape sh({1, 2, 3}, {1, 2, 3});
Tensor t = {sh, {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, kOnHost, ge::DT_FLOAT, nullptr};
t.SetDataType(ge::DT_DOUBLE);