SetFormat
函数功能
设置Tensor的数据排布格式。
函数原型
void SetFormat(Format format)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
format |
输入 |
数据排布格式,类型定义在cpu_types.h文件中。 |
返回值说明
无。
约束说明
无。
调用示例
shape->SetFormat(FORMAT_NCHW);
父主题: TensorShape类