Model Dictionaries

flood_forecast.model_dict_function.generate_square_subsequent_mask(sz: int) → torch.Tensor

Generate a square mask for the sequence. The masked positions are filled with float(‘-inf’). Unmasked positions are filled with float(0.0).