torch中与tensor相关的bug

1.expected type torch.FloatTensor but got torch.cuda.FloatTensor

查看了代码,是有个初始化的tensor是在cpu上的,相加的另一个tensor是在cuda上的。

修改:增加.cuda()

initial_hidden_state = initial_hidden_state.cuda()

猜你喜欢

转载自blog.csdn.net/u_7890/article/details/88880884