Shape构造函数
函数功能
Shape构造函数。
函数原型
下文中的dim_num_为维度个数,即有几维;dims_为具体的维度值信息。
- 默认构造一个shape,默认构造的shape实例中,dim_num_长度为0
- 通过dims_值构造shape,例如:Shape({8,3,224,224})表示创建一个Shape实例,有4个维度,每个维度的值分别是8,3,224,224
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
args |
输入 |
shape的所有dim值。 |
other |
输入 |
源shape对象。 |
返回值说明
生成一个初始化的Shape对象。
约束说明
无。
调用示例
Shape shape({3, 256, 256}); // dim_num_=3 dims_的前三维的维度为3,256,256
父主题: Shape