跳到主要内容

构造函数

函数功能

构造一个运行时shape实例。

函数原型

StorageShape()
StorageShape(const std::initializer_list<int64_t> &origin_shape, const std::initializer_list<int64_t> &storage_shape)

参数说明

参数输入/输出说明
origin_shape输入原始shape。
storage_shape输入运行时shape。

返回值

返回一个初始化后StorageShape对象。

约束说明

调用示例

StorageShape shape({3, 256, 256}, {3, 256, 256});