Class Unsqueeze

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.