Shape接口

接口功能

获取输出张量的维度值，以数组的形式返回。

接口格式

virtual const std::vector<int64_t> &Shape() const noexcept = 0;

接口参数

无。

使用样例

获取输出对象的张量维度信息。

1

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

返回值

返回输出张量维度数组。