跳到主要内容

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);