Tensor on GPU still getting error "expected type torch.cuda.FloatTensor"

Resolved. thanks!

RuntimeError: expected type torch.cuda.FloatTensor but got torch.FloatTensor