• Layer type: Flatten
• Doxygen Documentation
• Header: ./include/caffe/layers/flatten_layer.hpp
• CPU implementation: ./src/caffe/layers/flatten_layer.cpp

The Flatten layer is a utility layer that flattens an input of shape n * c * h * w to a simple vector output of shape n * (c*h*w).

• Parameters (FlattenParameter flatten_param)
• From ./src/caffe/proto/caffe.proto:

{% highlight Protobuf %} {% include proto/FlattenParameter.txt %} {% endhighlight %}