torch.amin

torch.amin(input, dim, keepdim=False, *, out=None) → Tensor

Возвращает минимальное значение каждого среза input тензора по заданному измерению(измерениям) dim.

Примечание

The difference between max/min and amax/amin is:

• amax/amin поддерживает уменьшение по нескольким измерениям,
• amax/amin не возвращает индексы,
• amax/amin равномерно распределяет градиент между одинаковыми значениями, в то время как max(dim)/min(dim) распространяет градиент только на один индекс в исходном тензоре.

Если keepdim равно True, тензор результата имеет те же размеры, что и input, за исключением измерения(измерений) dim, где он имеет размер 1. В противном случае, dim сжимается (см. torch.squeeze()), что приводит к тому, что тензор результата имеет на 1 (или len(dim)) меньше измерение(измерения).

Параметры:

• input (Tensor) – входной тензор.
• dim (int или кортеж из целых чисел) – измерение или измерения для уменьшения.
• keepdim (bool) – сохранять ли измерение в тензоре результата.

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

out (Tensor, необязательно) – тензор результата.

Пример:

>>> a = torch.randn(4, 4)
>>> a
tensor([[ 0.6451, -0.4866,  0.2987, -1.3312],
        [-0.5744,  1.2980,  1.8397, -0.2713],
        [ 0.9128,  0.9214, -1.7268, -0.2995],
        [ 0.9023,  0.4853,  0.9075, -1.6165]])
>>> torch.amin(a, 1)
tensor([-1.3312, -0.5744, -1.7268, -1.6165])