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 (Тензор) – входной тензор.
• dim (int или кортеж из целых чисел) – измерение или измерения для сокращения.
• keepdim (bool) – сохранить ли измерение в выходном тензоре.

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

out (Тензор, необязательно) – выходной тензор.

Пример:

>>> 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])