flatten

fun flatten(startDim: Int = 1, endDim: Int = -1): SequentialModelBuilder(source)

Adds a Flatten layer.