I think it just might!
Thanks a lot!
I might be able to fix this fiddling around with the shapes, but I am currently getting this error:
The size of tensor a (64) must match the size of tensor b (500) at non-singleton dimension 1