exp2_impl Class — pytorch Architecture

Source Code

aten/src/ATen/native/Math.h lines 1263–1269

template <typename T>
C10_HOST_DEVICE c10::complex<T> exp2_impl(c10::complex<T> x) {
  // There is no std::exp2 overload for complex, so instead
  // use the identity 2^x = e^(ln(2) * x)
  constexpr auto ln2 = c10::ln_2<T>;
  return std::exp(ln2 * x);
}