That is exactly what I wanted.  Why is list Comp allowed in this situation?  I just started pytorch so maybe I’m wrong but I thought any list comprehension would be running on the CPU.  If not, how is it running on the GPU?