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

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

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

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

torch.greater_equal(input, other)

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

Пример

import torch

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

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

Вывод:

tensor([True, True, True])

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

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