arange Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec512/vec512_complex_double.h lines 140–149

  template <typename step_t>
  static Vectorized<c10::complex<double>> arange(
      c10::complex<double> base = 0.,
      step_t step = static_cast<step_t>(1)) {
    return Vectorized<c10::complex<double>>(
        base,
        base + c10::complex<double>(1) * step,
        base + c10::complex<double>(2) * step,
        base + c10::complex<double>(3) * step);
  }