Class ExpandDims
Defined in File expand_dims.h
Inheritance Relationships
Base Type
public mindspore::ops::PrimitiveC
(Class PrimitiveC)
Class Documentation
-
class ExpandDims : public mindspore::ops::PrimitiveC
Adds an additional dimension to ‘input_x` at the given axis. Refer to Python API mindspore.ops.ExpandDims for more details.
Public Functions
-
inline ExpandDims()
Constructor.
-
~ExpandDims() = default
Destructor.
-
inline void Init()
Init. Refer to the parameters of Python API mindspore.ops.ExpandDims for the inputs.
-
inline ExpandDims()