Thanks for the reply!  Sorry I’m still a little new to pytorch but when you say I’m saving a reference to the CUDA tensor does that mean I should move it out of the GPU before I save it?  I’m trying to save all of those values so I can do some other work on them.