Функция PyTorch torch.less

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.lessЭто функция в PyTorch, используемая для поэлементного сравнения «меньше чем». Возвращает тензор, где элементы первого тензора меньше соответствующих элементов второго тензора в позициях, которые равны True.

Определение функции

torch.less(input, other)

Пример использования

Пример

import torch

# Создайте два тензора
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])

# Поэлементное сравнение «меньше чем»
result = torch.less(x, y)
print(result)

Результат вывода:

tensor([False, False, False])

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения