SetTensorShape

函数功能

设置Tensor的shape。

函数原型

bool SetTensorShape(const TensorShape *shape)

参数说明

返回值说明

bool类型。

• true：表示设置shape成功。
• false：表示失败。

约束说明

无。

调用示例

bool ret = tensor->SetTensorShape(shape);