SetDataSize
函数功能
设置Tensor的数据占用内存大小,单位字节。
函数原型
void SetDataSize(uint64_t size)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
size |
输入 |
Tensor的数据占用内存大小,单位字节。 |
返回值说明
无。
约束说明
无。
调用示例
tensor->SetDataSize(1024);
父主题: Tensor类