跳到主要内容

GetStorageShape

函数功能

获取运行时shape。

函数原型

const Shape &GetStorageShape() const

参数说明

返回值

运行时shape。

约束说明

调用示例

StorageShape shape({3, 256, 256}, {256, 256, 3});
auto storage_shape = shape.GetStorageShape(); // 256,256,3