I am pretty sure all these squeeze arguments would need the `axis=axis` argument, to not be trigger happy when there are other axes of dimension 1 around.