Функция torch.greater в PyTorch

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

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

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

torch.greater(input, other)

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

Пример

import torch

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

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

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

tensor([True, True, False])

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

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