squeeze

abstract fun squeeze(x: DTensor, axis: Int): DTensor