(self, backcast_size: int, forecast_size: int, interpolation_mode: str)
| 172 | # Cell |
| 173 | class _IdentityBasis(nn.Module): |
| 174 | def __init__(self, backcast_size: int, forecast_size: int, interpolation_mode: str): |
| 175 | super().__init__() |
| 176 | assert (interpolation_mode in ['linear','nearest']) or ('cubic' in interpolation_mode) |
| 177 | self.forecast_size = forecast_size |
| 178 | self.backcast_size = backcast_size |
| 179 | self.interpolation_mode = interpolation_mode |
| 180 | |
| 181 | def forward(self, theta: t.Tensor, insample_x_t: t.Tensor, outsample_x_t: t.Tensor) -> Tuple[t.Tensor, t.Tensor]: |
| 182 |