Hi all,
How to set ‘Inf’ in Tensor to 0?
I don’t wish to use numpy since that require to set backward when using it in Networks.
Thanks,
Qinqing
Hi all,
How to set ‘Inf’ in Tensor to 0?
I don’t wish to use numpy since that require to set backward when using it in Networks.
Thanks,
Qinqing
x = torch.Tensor([1, float("Inf"), 2, float("Inf")])
x[x == float("Inf")] = 0
x # should be 1, 0, 2, 0 now
Thanks for the answer! May I know if the masked entry will affect the gradient? For example, if I have a model whose intermediate layer gives an output of:
out = [inf, -3.4, inf, -5.5]
where the inf
entries should be masked out, can I do it by:
out[out == float('inf')] = 0
Thanks!