How to i represent three lists in one tensor state for a DQN

I am building a DQN for atari game playing, and i have an algorithm that gives me data about objects in each frame, which are represented as three lists, first is the X-coordinate of an object second is the y-coordinate, and third is what class the object is in. an example would look like this:

X=[22.3,54.0,1.12] Y=54.3,23.5,126.5] class=[1,1,2]

i am intentionally using handcrafted methods rather than a CNN for my final year dissertation, and this implementation is using pytorch libraries in python.

my problem is in trying to transform this multidimensional data into one torch variable that can be used as the state for the DQN, as i don't think i can simply make them a list of torch stacks.

Any suggestions would be appreciated