AIfES 2  2.0.0
ailayer_flatten Struct Reference

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.
 

Detailed Description

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.


The documentation for this struct was generated from the following file: