SetExpandDimsType
Function Usage
Sets dimension expansion rules.
Prototype
1 | void SetExpandDimsType(const ExpandDimsType &expand_dims_type) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
expand_dims_type |
Input |
Dimension expansion rule |
Returns
None
Constraints
None
Examples
1 2 3 4 | 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