I came across this problem recently and didn’t quite get what you meant because UVs are only 2-dimensinal, so I only had the X and Y position. But after I set the “Num custom UVs” to 2 (in the Material settings) I was able to get the Z axis as well.
