Shape接口

接口功能

获得推理输入张量的维度信息。

接口格式

const std::vector<int64_t> &Shape() const;

接口参数

无。

使用样例

获取输入数据维度信息。

1

std::vector<int64_t> shape = input->Shape();

返回值

输入张量的维度信息。