flatten

fun DagBuilder.flatten(tensor: GraphValue<*>, startDim: Int = 0, endDim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for flatten.