昇腾社区首页
中文
注册

MutableTensorData

函数功能

获取tensor中的数据。

函数原型

TensorData &MutableTensorData()

参数说明

无。

返回值说明

可写的tensor data引用。

关于TensorData类型的定义,请参见TensorData

约束说明

无。

调用示例

StorageShape sh({1, 2, 3}, {1, 2, 3});
Tensor t = {sh, {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, kOnHost, ge::DT_FLOAT, nullptr};
auto a = reinterpret_cast<void *>(10);
t.MutableTensorData() = TensorData{a, nullptr}; // 设置新tensordata
auto td = t.GetTensorData(); // TensorData{a, nullptr}