Equivalent to `dx = dx[(slice(None),) + (None,)*(y.ndim-1)]` ?