_nan_to_num_replace Class — pytorch Architecture

Source Code

aten/src/ATen/native/cpu/UnaryOpsKernel.cpp lines 446–458

template <typename scalar_t>
inline scalar_t _nan_to_num_replace(
    scalar_t a, scalar_t nan_replacement, scalar_t pos_inf_replacement, scalar_t neg_inf_replacement) {
  if (at::_isnan(a)) {
    return nan_replacement;
  } else if (a == std::numeric_limits<scalar_t>::infinity()) {
    return pos_inf_replacement;
  } else if (a == -std::numeric_limits<scalar_t>::infinity()) {
    return neg_inf_replacement;
  } else {
    return a;
  }
}