创建掩码

pytorch_forecasting.utils._utils.create_mask(size: int, lengths: LongTensor, inverse: bool = False) → BoolTensor

创建形状为 len(lenghts) x size 的布尔掩码。

如果 lengths[i] > j，则 (i, j) 处的条目为 True。

Parameters:

• size (int) – 第二维的大小

• lengths (torch.LongTensor) – 长度的张量

• 逆转 (布尔值, 可选) – 如果为真，则布尔遮罩被反转。默认为假。

Returns:

掩码

Return type:

torch.BoolTensor