Функция torch.allclose в PyTorch
Справочное руководство PyTorch torch
torch.allcloseФункция в PyTorch, используемая для проверки того, близки ли все элементы двух тензоров. Она используется для определения того, равны ли два тензора в пределах заданного допуска.
Определение функции
torch.allclose(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.allclose(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.allclose(x, y)
print(result)
Результат вывода:
True
Другие расширения