arange Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/vsx/vec256_int32_vsx.h lines 166–179

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