![]() |
AIfES 2
2.0.0
|
Structure of the Flatten layer, based on the reshape layer . More...
#include <ailayer_reshape.h>
Data Fields | |
| ailayer_reshape_t | base |
| Inherited field members from reshape layer struct. | |
| uint16_t | output_shape [2] |
| Target 2D shape for flatten operation. | |
Structure of the Flatten layer, based on the reshape layer .
This layer is used to convert a ND tensor to a 2D tensor, for example as an input for a Dense layer.