Функция torch.equal в PyTorch
Справочное руководство Pytorch torch
torch.equalЭто функция в PyTorch, используемая для проверки равенства двух тензоров. Если два тензора имеют одинаковую форму и элементы, она возвращает True.
Определение функции
torch.equal(input, other)
Пример использования
Пример
import torch
# Создать два одинаковых тензора
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Проверить, равны ли они
result = torch.equal(x, y)
print(result)
# Создать разные тензоры
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
# Создать два одинаковых тензора
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# Проверить, равны ли они
result = torch.equal(x, y)
print(result)
# Создать разные тензоры
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
Результат вывода:
True False
Другие расширения