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]