SetExpandDimsType

函数功能

设置补维规则。

函数原型

void SetExpandDimsType(const ExpandDimsType &expand_dims_type)

参数说明

返回值说明

无。

约束说明

无。

调用示例

ExpandDimsType dim_type("1100");
StorageFormat format(ge::Format::FORMAT_NCHW, ge::Format::FORMAT_C1HWNC0, dim_type);
ExpandDimsType new_dim_type("1010");
format.SetExpandDimsType(new_dim_type);