分组应用

pytorch_forecasting.utils._utils.groupby_apply(keys: Tensor, values: Tensor, bins: int = 95, reduction: str = 'mean', return_histogram: bool = False) → Tensor | Tuple[Tensor, Tensor]

对torch张量进行分组应用

Parameters:

• keys – 组的张量 (0 到 bins)

• values – 要聚合的值 - 与键的大小相同

• bins – 总组数

• reduction – 可选“mean”或“sum”

• return_histogram – 是否返回顶部的直方图

Returns:

大小为 bins 的张量，包含汇总值，并可选地带有值的计数