Class Unsqueeze
Defined in File unsqueeze.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class Unsqueeze : public mindspore::ops::BaseOperator
Unsqueeze defined the Unsqueeze operator prototype of lite.
Public Functions
-
inline Unsqueeze()
Constructor.
-
void Init(const std::vector<int64_t> axis)
Method to init the op’s attributes.
- Parameters
axis – [in] Define a vector to indicate on which dimensions to expand.
-
void set_axis(const std::vector<int64_t> axis)
Method to set axis attribute.
- Parameters
axis – [in] Define a vector to indicate on which dimensions to expand.
-
std::vector<int64_t> get_axis() const
Method to get axis attribute.
- Returns
dimensions info of expanding.
-
inline Unsqueeze()