Method Intersect

oneflow/core/register/tensor_slice_view.cpp:75–90

}

TensorSliceView TensorSliceView::Intersect(const TensorSliceView& other) const {
  if (IsEmpty() || other.IsEmpty()) { return TensorSliceView(); }
  CHECK_EQ(other.range_vec_.size(), range_vec_.size());
  std::vector<Range> intersection_vec;
  intersection_vec.reserve(range_vec_.size());
  const Range zero(0, 0);
  FOR_RANGE(int64_t, i, 0, range_vec_.size()) {
    const Range intersection = FindIntersectant(range_vec_.at(i), other.range_vec_.at(i));
    if (intersection == zero) {
      return TensorSliceView();
    } else {
      intersection_vec.emplace_back(intersection);
    }
  }
  return TensorSliceView(intersection_vec);
}

const Range& TensorSliceView::At(int64_t index) const { return range_vec_.at(index); }