Class FlattenLayer

java.lang.Object
All Implemented Interfaces:
Layer<Tensor1D>, Serializable

public class FlattenLayer extends AbstractLayer<Tensor1D,Tensor1D,TensorBase>
Transforms outputs from previous 3D layer into a flatten 1D tensor in forward pass, Backward pass propagates weighted errors/deltas from the next fully connected layer. Automatically added after 2D or 3D layer to transition to fully connected layers.
See Also: