create_mask#

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

Create boolean masks of shape len(lengths) x size.

An entry at (i, j) is True if lengths[i] > j.

Parameters:

  • size (int) – Size of second dimension.

  • lengths (torch.LongTensor) – Tensor of lengths.

  • inverse (bool, optional) – If true, boolean mask is inverted. Defaults to False.

Returns:

Mask tensor.

Return type:

torch.BoolTensor