See https://github.com/astropy/astropy/issues/6778 for how to do complex numbers (spoiler: use the 0th dim for real/imaginary).