shape

shape用于定义张量的逻辑形状,用于描述各维度的大小。

头文件路径为:tensor_api/tensor.h

原型定义

template <typename... Shapes>
using shape = Std::tuple<Shapes...>;

参数说明

表1 参数说明

参数名 输入/输出 描述
Shapes... 输入 可变参数模板,表示各维度的形状值,也可表示嵌套的shape子结构。

API映射关系

shape通过make_shape函数创建,通过get_shape函数和layout的成员shape获取。