Flatten - SkyWorld117/DianoiaML.jl GitHub Wiki

Flatten, a flatten-layer.

Flatten(;input_shape::Tuple)

input_shape: a tuple of input data shape, can be set automatically, unless it is the first layer in the network