SetExpandDimsType
Description
Sets dimension expansion rules.
Prototype
void SetExpandDimsType(const ExpandDimsType &expand_dims_type)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
expand_dims_type |
Input |
Dimension expansion rule |
Returns
None
Restrictions
None
Example
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);
Parent topic: StorageFormat