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

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

torch.maximumЭто функция в PyTorch для поэлементного максимума. Возвращает новый тензор, составленный из максимумов соответствующих элементов двух тензоров.

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

torch.maximum(input, other)

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

Пример

import torch

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

# Вычислить поэлементный максимум
result = torch.maximum(x, y)
print(result)

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

tensor([3, 5, 4])

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

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