Функция PyTorch torch.ge

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

torch.ge— это функция в PyTorch, используемая для поэлементного сравнения на «больше или равно».

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

torch.ge(input, other, out)

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

Пример

import torch

# Поэлементное сравнение на «больше или равно»
x = torch.tensor([1, 2, 3, 4])
y = torch.tensor([2, 2, 2, 5])

result = torch.ge(x, y)
print(result)

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

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