torch.isclose

torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) → Tensor

Возвращает новый тензор с булевыми элементами, представляющими, является ли каждый элемент input «близким» к соответствующему элементу other. Близость определяется как:

$$\lvert \text{input} - \text{other} \rvert \leq \texttt{atol} + \texttt{rtol} \times \lvert \text{other} \rvert$$

где input и other являются конечными. Где input и/или other не являются конечными, они близки тогда и только тогда, когда они равны, при этом NaN считаются равными друг другу, когда equal_nan равно True.

Параметры

• input (Тензор) – первый тензор для сравнения
• other (Тензор) – второй тензор для сравнения
• atol (float, необязательно) – абсолютная погрешность. По умолчанию: 1e-08
• rtol (float, необязательно) – относительная погрешность. По умолчанию: 1e-05
• equal_nan (bool, необязательно) – если True, то два NaN будут считаться равными. По умолчанию: False

Примеры:

>>> torch.isclose(torch.tensor((1., 2, 3)), torch.tensor((1 + 1e-10, 3, 4)))
tensor([ True, False, False])
>>> torch.isclose(torch.tensor((float('inf'), 4)), torch.tensor((float('inf'), 6)), rtol=.5)
tensor([True, True])