Class FlattenLayer

All Implemented Interfaces:
Backward, Forward, Layer<TensorBase>, Serializable

public class FlattenLayer extends AbstractLayer<TensorBase, TensorBase, 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: