has_nan_or_inf

Contents

has_nan_or_inf#

ptyrad.utils.dev_tools.has_nan_or_inf(tensor)[source]#

Check if a torch.Tensor contains any NaN or Inf values.

Parameters:

tensor (torch.Tensor) – Input tensor to check.

Returns:

True if the tensor contains any NaN or Inf values, False otherwise.

Return type:

bool