Функция PyTorch torch.isclose
Справочное руководство Pytorch torch
torch.iscloseЭто функция в PyTorch, используемая для проверки, близки ли два тензора. Возвращает булев тензор, указывающий, близки ли соответствующие элементы в двух тензорах в пределах заданного допуска.
Определение функции
torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)
Пример использования
Пример
import torch
# Создайте два похожих тензора
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Проверить, близки ли соответствующие элементы
result = torch.isclose(x, y)
print(result)
# Создайте два похожих тензора
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# Проверить, близки ли соответствующие элементы
result = torch.isclose(x, y)
print(result)
Результат вывода:
tensor([True, True, True])
Другие расширения