检查非有限值#
- pytorch_forecasting.data.timeseries.check_for_nonfinite(tensor: Tensor, names: str | List[str]) Tensor [来源]#
检查张量是否包含NAs或无限值,并且具有正确的维度。
检查:
判断张量是否有限,否则引发ValueError
检查张量的维度是否正确。如果张量是字符串,tensor.ndim 必须为 1;如果张量是列表,tensor.ndim 必须为 2。否则会引发 AssertionError。
- Parameters:
names (str 或 list of str) – 要检查的列的名称
张量 (torch.Tensor) – 要检查的张量
- Returns:
如果检查没有问题,则返回未更改的张量
- Return type:
torch.Tensor
- Raises:
ValueError – 如果张量包含NAs或无限值
AssertionError – 如果张量的维度不正确,请参见上文