pkgs/development/libraries/science/math/libtorch/test/test.cpp at 20.09

nixpkgs mirror (for testing) github.com/NixOS/nixpkgs

nix

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

at 20.09 7 lines 137 B view raw

wrap content

1#include <torch/torch.h>
2#include <iostream>
3
4int main() {
5  torch::Tensor tensor = torch::eye(3);
6  std::cout << tensor << std::endl;
7}