跳到主要内容

SetSize

函数功能

设置Tensor的内存大小。

函数原型

void SetSize(const size_t size)

参数说明

参数输入/输出说明
size输入Tensor的内存大小,单位是字节。

返回值

约束说明

调用示例

StorageShape sh({1, 2, 3}, {1, 2, 3});
Tensor t = {sh, {}, {}, ge::DT_FLOAT, nullptr};
t.SetSize(0U);