nixpkgs/pkgs/development/libraries/science/math/libtorch/test/test.cpp

#include <torch/torch.h>
#include <iostream>

int main() {
  torch::Tensor tensor = torch::eye(3);
  std::cout << tensor << std::endl;
}