torch.aminmax

torch.aminmax(input, *, dim=None, keepdim=False, out=None) -> (Tensor min, Tensor max)

Вычисляет минимальное и максимальное значения input тензора.

Параметры:

input (Тензор) – Входной тензор

Ключевые аргументы:

• dim (Необязательный[int]) – Размерность, по которой вычисляются значения. Если None, вычисляет значения для всего input тензора. Значение по умолчанию None.
• keepdim (bool) – Если True, уменьшенные размеры будут сохранены в выходном тензоре в виде размеров с размером 1 для трансляции, в противном случае они будут удалены, как если бы вызывали (torch.squeeze()). Значение по умолчанию False.
• out (Необязательный[Кортеж[Тензор, Тензор]]) – Необязательные тензоры, в которые будет записан результат. Должны иметь такую же форму и тип данных, как ожидаемый выход. Значение по умолчанию None.

Возвращает:

Именованная кортеж (min, max) содержащий минимальное и максимальное значения.

Возбуждает:

RuntimeError – Если размерность для вычисления значений имеет размер 0.

Примечание

Значения NaN передаются в выход, если по крайней мере одно значение NaN.

См. также

torch.amin() вычисляет только минимальное значение torch.amax() вычисляет только максимальное значение

Пример:

>>> torch.aminmax(torch.tensor([1, -3, 5]))
torch.return_types.aminmax(
min=tensor(-3),
max=tensor(5))

>>> # aminmax propagates NaNs
>>> torch.aminmax(torch.tensor([1, -3, 5, torch.nan]))
torch.return_types.aminmax(
min=tensor(nan),
max=tensor(nan))

>>> t = torch.arange(10).view(2, 5)
>>> t
tensor([[0, 1, 2, 3, 4],
        [5, 6, 7, 8, 9]])
>>> t.aminmax(dim=0, keepdim=True)
torch.return_types.aminmax(
min=tensor([[0, 1, 2, 3, 4]]),
max=tensor([[5, 6, 7, 8, 9]]))