taco: add tensor mode shifting logic to c++

Author: rohany

taco/bench.h

@@ -40,4 +40,20 @@
 
 taco::TensorBase loadRandomTensor(std::string name, std::vector<int> dims, float sparsity, taco::Format format);
 
+template<typename T>
+taco::Tensor<T> shiftLastMode(std::string name, taco::Tensor<T> original) {
+  taco::Tensor<T> result(name, original.getDimensions(), original.getFormat());
+  std::vector<int> coords(original.getOrder());
+  for (auto& value : taco::iterate<T>(original)) {
+    for (int i = 0; i < original.getOrder(); i++) {
+      coords[i] = value.first[i];
+    }
+    int lastMode = original.getOrder() - 1;
+    coords[lastMode] = (coords[lastMode] + 1) % original.getDimension(lastMode);
+    result.insert(coords, value.second);
+  }
+  result.pack();
+  return result;
+}
+
 #endif //TACO_BENCH_BENCH_H