flatten

fun flatten(a: Tensor<*, *>, startDim: Int = 0, endDim: Int = -1): Tensor<*, *>(source)

Flatten the tensor between startDim and endDim.