torch.argmin

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

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

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

Примечание

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

Параметры:

• input (Tensor) – входной тензор.
• 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]])