Class Squeeze

Defined in File squeeze.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class Squeeze : public mindspore::ops::BaseOperator 

Returns a tensor with the same data type but dimensions of 1 are removed based on axis. Refer to Python API mindspore.ops.Squeeze for more details.

Public Functions

inline Squeeze(): Constructor.

void Init(const std::vector<int64_t> &axis = {}): Init. Refer to the parameters of Python API mindspore.ops.Squeeze for the inputs.

void set_axis(const std::vector<int64_t> &axis): Set axis.

std::vector<int64_t> get_axis() const

Get axis.

返回: axis.