跳到主要内容

GetData

函数功能

获取Tensor的数据地址。

函数原型

template<class T> const T *GetData() const
template<class T> T *GetData()

参数说明

参数输入/输出说明
T输入数据类型。

返回值

数据地址。

约束说明

调用示例

Tensor tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}}, // shape
{ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, // format
kFollowing, // placement
ge::DT_FLOAT16, // dt
nullptr};
auto addr = tensor.GetData<int64_t>();