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.