Получите ошибку, когда задом наперед: «RuntimeError: Tensor: неверное смещение хранилища»? - PullRequest
0 голосов
/ 06 мая 2019

Когда я использую 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()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...