def get_step(v, dv):
I = dv < 1e-12
if torch.sum(I) > 0: # TODO: Use something like torch.any(dv < 0)
a = -v / dv
return torch.min(a[I])
else:
return 1
微信
公众号
扫码关注公众号