squeeze

fun squeeze(a: Tensor<*, *>, dim: Int? = null): Tensor<*, *>(source)

Remove dimensions of size 1.