跳到主要内容

MutableOriginShape

函数功能

获取可写的原始shape。

函数原型

Shape &MutableOriginShape()

参数说明

返回值

可写的原始shape。

约束说明

调用示例

StorageShape shape({3, 256, 256}, {256, 256, 3});
auto origin_shape = shape.MutableOriginShape(); // 3,256,256