FlattenLayer

FlattenLayer[] represents a net layer that flattens any input array into a vector.

• FlattenLayer[n] represents a net layer that flattens its input to level n.
• FlattenLayer[{{s11, s12, ...}, {s21, s22, ...}, ...}] represents a net layer that flattens its input by combining all levels sij to make each level i in the result.

Examples

Create a flatten layer:

FlattenLayer[]

Apply a flatten layer to an array:

layer = FlattenLayer[];
layer[{{1, 2}, {3, 4}}, NetEvaluationMode -> "Train"]

Please visit the official Wolfram Language Reference for more details.