设置Tensor的数据占用内存大小,单位字节。
void SetDataSize(uint64_t size)
参数
输入/输出
说明
size
输入
Tensor的数据占用内存大小,单位字节。
无。
tensor->SetDataSize(1024);