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获取。