arange Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec512/vec512_complex_float.h lines 627–640

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