设置Tensor的数据。
void SetData(TensorData &&data)
参数 |
输入/输出 |
说明 |
---|---|---|
data |
输入 |
需要设置的数据。 关于TensorData类型的定义,请参见TensorData。 |
无。
无。
Tensor t{{{8, 3, 224, 224}, {16, 3, 224, 224}}, // shape {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, // format kOnHost, // placement ge::DT_FLOAT16, //dt nullptr}; void *a = &t; TensorData td(a, nullptr); t.SetData(std::move(td));