SetExpandDimsType
Description
Sets the dimension expansion rule of a shape.
Prototype
void SetExpandDimsType(const ExpandDimsType &expand_dims_type)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
expand_dims_type |
Input |
Dimension expansion rule to be set. For details about the definition of the ExpandDimsType type, see ExpandDimsType. |
Returns
None
Restrictions
None
Example
Tensor tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}}, // shape
{ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, // format
kFollowing, // placement
ge::DT_FLOAT16, //dt
nullptr};
ExpandDimsType type("1001");
tensor.SetExpandDimsType(type);
auto expand_dims_type = tensor.GetExpandDimsType(); // type
Parent topic: Tensor