向后获取错误:“ RuntimeError:张量:无效的存储偏移量”?

时间:2019-05-06 13:18:59

标签: pytorch

当我使用pytorch进行向后毕业时,出现错误:

RuntimeError: Tensor: invalid storage offset at c:\users\administrator\downloads\new-builder\win-wheel\pytorch\aten\src\th\generic/THTensor.cpp:761

H_grad = torch.autograd.grad(outputs=prediction.sum(), inputs=database_choose, create_graph=True)[0]
Ht=H_grad[:,1].reshape(choose,1)
Htt=torch.autograd.grad(outputs=Ht.sum(), inputs=database_choose,create_graph=True)[0][:,1].reshape(choose,1)
loss.backward()

0 个答案:

没有答案