Shape构造函数

函数功能

shape构造函数，可以默认构造一个dim长度为0的shape实例，也可以构造一个通过dims值指定的shape。例如，您可以通过Shape({8,3,224,224})创建一个shape实例，则此shape有4个维度，每个维度的值分别为8、3、224、224。

函数原型

Shape() // 默认构造函数，生成维度为0的Shape对象

Shape(std::initializer_list<int64_t> args) : dim_num_(0), dims_{0}

参数说明

返回值说明

生成一个初始化的Shape对象，维度为args的size，tensor的shape为args的值。

约束说明

无。

调用示例

Shape shape({3, 256, 256}); // dim_num_=3  dims_的前三维的维度为3,256,256