检查非有限值#

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 (strlist of str) – 要检查的列的名称

  • 张量 (torch.Tensor) – 要检查的张量

Returns:

如果检查没有问题,则返回未更改的张量

Return type:

torch.Tensor

Raises:
  • ValueError – 如果张量包含NAs或无限值

  • AssertionError – 如果张量的维度不正确,请参见上文