mask Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec_n.h lines 123–133

  template <int64_t mask>
  static VectorizedN<T, N> blend(
      const VectorizedN<T, N>& a,
      const VectorizedN<T, N>& b) {
    VectorizedN<T, N> result;
    for (int i = 0; i < N; ++i) {
      result.values[i] =
          Vectorized<T>::template blend<mask>(a.values[i], b.values[i]);
    }
    return result;
  }