masked_op

pytorch_forecasting.utils._utils.masked_op(tensor: Tensor, op: str = 'mean', dim: int = 0, mask: Tensor = None) → Tensor

对掩码张量进行计算操作。

Parameters:

• 张量 (torch.Tensor) – 用于进行操作的张量

• op (str) – 要应用的操作。可以是 [“mean”, “sum”] 中的一种。默认为 “mean”。

• dim (int, optional) – 用于求平均值的维度。默认为 0。

• mask (torch.Tensor, 可选) – 要应用的布尔掩码（True=取平均，False=忽略）。默认情况下屏蔽nan值。

Returns:

经过平均的维度的张量

Return type:

torch.Tensor