Class Flatten

Defined in File flatten.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class Flatten : public mindspore::ops::BaseOperator 

Flattens a tensor without changing its batch size on the 0-th axis. Refer to Python API mindspore.ops.Flatten for more details.

Public Functions

inline Flatten(): Constructor.

void Init(const int64_t axis = 1): Init. Refer to the parameters of Python API mindspore.ops.Flatten for the inputs.

void set_axis(const int64_t axis): Set axis.

int64_t get_axis() const

Get axis.

返回: axis.