If you create a Variable in the forward(), it cannot switch to cuda automatically. you must specify .cuda() . Something like:
h = Variable(torch.randn(2, batch_size, 256*2))
h = h.cuda()
My English is poor. Hope it can help you.
中国人？
如果你在forward() 里定义一个Variable,想让它在cuda里运行的话， 你的担忧是对的，它不会自动的转到cuda里，应该需要.cuda()。就我所了解的应该是这样。要不然的话估计会报错？我没有试过。
我也是新手，说错了不要怪我啊，相互学习！