torch.quantile

torch.quantile(input, q, dim=None, keepdim=False, *, interpolation='linear', out=None) → Tensor

Вычисляет q-е квантили каждой строки input тензора вдоль измерения dim.

Для вычисления квантиля мы отображаем q в [0, 1] на диапазон индексов [0, n], чтобы найти расположение квантиля в отсортированном входном массиве. Если квантиль находится между двумя точками данных a < b с индексами i и j в отсортированном порядке, результат вычисляется в соответствии с заданным interpolation методом следующим образом:

• linear: a + (b - a) * fraction, где fraction — дробная часть вычисленного индекса квантиля.
• lower: a.
• higher: b.
• nearest: a или b, чьей индекс ближе к вычисленному индексу квантиля (округляя вниз для дробей .5).
• midpoint: (a + b) / 2.

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

Примечание

По умолчанию dim является None в результате чего тензор input сжимается перед вычислением.

Параметры:

• input (Tensor) – входной тензор.
• q (float или Tensor) – скаляр или одномерный тензор со значениями в диапазоне [0, 1].
• dim (int) – измерение для сокращения.
• keepdim (bool) – сохраняется ли измерение в выходном тензоре или нет.

Аргументы ключевого слова:

• interpolation (str) – метод интерполяции, используемый, когда искомый квантиль лежит между двумя точками данных. Может быть linear, lower, higher, midpoint и nearest. По умолчанию linear.
• out (Tensor, необязательно) – выходной тензор.

Пример:

>>> a = torch.randn(2, 3)
>>> a
tensor([[ 0.0795, -1.2117,  0.9765],
        [ 1.1707,  0.6706,  0.4884]])
>>> q = torch.tensor([0.25, 0.5, 0.75])
>>> torch.quantile(a, q, dim=1, keepdim=True)
tensor([[[-0.5661],
        [ 0.5795]],

        [[ 0.0795],
        [ 0.6706]],

        [[ 0.5280],
        [ 0.9206]]])
>>> torch.quantile(a, q, dim=1, keepdim=True).shape
torch.Size([3, 2, 1])
>>> a = torch.arange(4.)
>>> a
tensor([0., 1., 2., 3.])
>>> torch.quantile(a, 0.6, interpolation='linear')
tensor(1.8000)
>>> torch.quantile(a, 0.6, interpolation='lower')
tensor(1.)
>>> torch.quantile(a, 0.6, interpolation='higher')
tensor(2.)
>>> torch.quantile(a, 0.6, interpolation='midpoint')
tensor(1.5000)
>>> torch.quantile(a, 0.6, interpolation='nearest')
tensor(2.)
>>> torch.quantile(a, 0.4, interpolation='nearest')
tensor(1.)