torch.isclose

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

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

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

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

Параметры:

• input (Tensor) – первый тензор для сравнения
• other (Tensor) – второй тензор для сравнения
• rtol (float, optional) – относительный допуск. Значение по умолчанию: 1e-05
• atol (float, optional) – абсолютный допуск. Значение по умолчанию: 1e-08
• equal_nan (bool, optional) – если 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])