查询 PyTorch 中任意点离散分布的 CDF

问：

我有一个离散分布说.CDF 由 给出。这里的 和 数组表示给定 处的相应值。就我而言，它实际上不是一个离散分布，而是带有支撑的分段均匀分布，即我想在不同点查询该分布的 CDF，比如 .现在。有没有办法（一些库函数）无需手动计算即可快速获取？x=[1,2,4,5]; P(x)=[0.1, 0.2, 0.3, 0.4]F(x)=[0.1, 0.3, 0.6, 1]PFxPx \in [1, 9]x \in [1, 2], P(x)=0.1; x \in [2, 4], P(x)=0.1; x \in [4, 5], P(x)=0.3; x \in [5,9], P(x)=0.1y=[0.5, 1.6, 2.3, 3.4, 4.5, 5.7, 8.9]F(y)=[0, 0.06, 0.13, 0.24, 0.45, 0.67, 0.99]F(y)

在我的例子中，并且是形状的值列表（2D 数组）的数组，也是形状的 2D 数组。我想在 PyTorch 中有效地计算这一点，这也应该允许梯度反向传播。xP(batch_size, num_bins)y(batch_size, num_new_bins)