[complex] torch.abs: does not match numpy #48486
Description
import torch
import numpy as np
print('torch.cfloat\n')
x = torch.tensor([1e20+1e10j], dtype=torch.cfloat)
print(torch.abs(x), np.abs(x.numpy())) # non-vectorized: correct value
x = torch.tensor([1e20+1e10j] * 10, dtype=torch.cfloat)
print(torch.abs(x)) # vectorized: incorrect value
print(np.abs(x.numpy()))
print('\ntorch.cdouble\n')
x = torch.tensor([1e200+0j], dtype=torch.cdouble)
print(torch.abs(x), np.abs(x.numpy())) # non-vectorized: correct value
x = torch.tensor([1e200+0j] * 10, dtype=torch.cdouble)
print(torch.abs(x)) # vectorized: incorrect value
print(np.abs(x.numpy()))Output
torch.cfloat
tensor([1.0000e+20]) [1.e+20]
tensor([ inf, inf, inf, inf, inf, inf,
inf, inf, 1.0000e+20, 1.0000e+20])
[1.e+20 1.e+20 1.e+20 1.e+20 1.e+20 1.e+20 1.e+20 1.e+20 1.e+20 1.e+20]
torch.cdouble
tensor([1.0000e+200], dtype=torch.float64) [1.e+200]
tensor([ inf, inf, inf, inf, inf,
inf, inf, inf, 1.0000e+200, 1.0000e+200],
dtype=torch.float64)
[1.e+200 1.e+200 1.e+200 1.e+200 1.e+200 1.e+200 1.e+200 1.e+200 1.e+200
1.e+200]
cc @ezyang @anjali411 @dylanbespalko @mruberry @rgommers @heitorschueroff