Method OffsetTo

oneflow/core/register/tensor_slice_view.cpp:100–107

size_t TensorSliceView::NumAxes() const { return range_vec_.size(); }

NdIndex TensorSliceView::OffsetTo(const TensorSliceView& other) const {
  CHECK_EQ(other.NumAxes(), NumAxes());
  DimVector indices_vec(range_vec_.size());
  std::transform(range_vec_.cbegin(), range_vec_.cend(), other.range_vec_.cbegin(),
                 indices_vec.begin(),
                 [](const Range& lhs, const Range& rhs) { return lhs.begin() - rhs.begin(); });
  return NdIndex(indices_vec);
}

void TensorSliceView::ToProto(TensorSliceViewProto* proto) const {
  for (const Range& range : range_vec_) { range.ToProto(proto->mutable_dim()->Add()); }