xtensor-simple-example.cpp

#include <iostream>

#include "xtensor/xio.hpp"
#include "xtensor/xrandom.hpp"
#include "xtensor/xtensor.hpp"
#include "xtensor/xview.hpp"

template <class E>
auto transform(E& x) {
  x *= 2;
}

using farray = xt::xtensor<float, 2, xt::layout_type::column_major>;

int main() {
  const farray x_orig = xt::random::randn<float>({2, 2});
  using namespace xt::placeholders;

  farray x_view = x_orig;
  auto x1 = xt::view(x_view, xt::all(), xt::range(_, 1));
  transform(x1);
  auto x2 = xt::view(x_view, xt::all(), xt::range(1, _));
  transform(x2);
  std::cout << "Same with views? " << xt::allclose(x_view, 2 * x_orig)
            << std::endl;

  std::cout << 2 * x_orig << std::endl;
  std::cout << "--------------------" << std::endl;
  std::cout << x_view << std::endl;

  return 0;
}