torch.argmin

torch.argmin(input, dim=None, keepdim=False) → LongTensor

Возвращает индексы минимального значения (значений) сплющенного тензора или по размерности

Это второе значение, возвращаемое функцией torch.min(). См. её документацию для точного определения семантики этого метода.

Примечание

Если существует несколько минимальных значений, возвращаются индексы первого минимального значения.

Параметры

• input (Тензор) – входной тензор.
• dim (int) – размерность для сокращения. Если None, возвращается argmin сплющенного входного тензора.
• keepdim (bool) – сохраняет ли выходной тензор размерность или нет.

Пример:

>>> a = torch.randn(4, 4)
>>> a
tensor([[ 0.1139,  0.2254, -0.1381,  0.3687],
        [ 1.0100, -1.1975, -0.0102, -0.4732],
        [-0.9240,  0.1207, -0.7506, -1.0213],
        [ 1.7809, -1.2960,  0.9384,  0.1438]])
>>> torch.argmin(a)
tensor(13)
>>> torch.argmin(a, dim=1)
tensor([ 2,  1,  3,  1])
>>> torch.argmin(a, dim=1, keepdim=True)
tensor([[2],
        [1],
        [3],
        [1]])