pytorch_forecasting.utils.create_mask

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

Create boolean masks of shape len(lenghts) 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

Return type

torch.BoolTensor

pytorch_forecasting.utils.autocorrelation pytorch_forecasting.utils.get_embedding_size