pytorch loss inf

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/82932569

import torch

a=torch.zeros(1)
b=torch.ones(1,requires_grad=True)
print(b/0)

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/82932569
inf
今日推荐