torch.quantile or np.quantile的计算

torch的文档详细说明了quantile的计算方法。

主要是将q的范围[0, 1] 转成输入index的范围[0, n]。 也就是说，将q 乘 n。然后插值计算。

quantile位置不是整数

a = torch.tensor([0.0, 1.1, 2.1, 3.1])
q_result = torch.quantile(a, torch.tensor([0.1, 0.9]))
# tensor([0.3300, 2.8000])

a的index范围： [0, 3]
两个q值0.1, 0.9 都乘以3之后得到0.3， 2.7。

• 0.3在0和1之间，需要插值。缺省插值方法为a + (b-a) * fraction。

0.0+ （1.1 - 0.0）*0.3 = 0.3300

• 2.7在2和3之间

2.1 + (3.1 - 2.1 ) * 0.7 = 2.8000

quantile位置为整数

不需要插值。
例如

a = torch.tensor([0.1, 0.2, 0.3, 0.33, 0.9])
q2 = torch.quantile(a, torch.tensor([0.5]))
# 0.3 

0.5 * 4 = 2， 因此取a[2]