Class Flatten

• Defined in File flatten.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Class Documentation

class Flatten : public mindspore::ops::PrimitiveC

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.

~Flatten() = default

Destructor.

inline void Init()

Init. Refer to the parameters of Python API mindspore.ops.Flatten for the inputs.