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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

#undef NDEBUG
#include <cassert>

#include <iostream>

#include <torch/torch.h>

int main() {
  torch::Tensor tensor = torch::eye(3);

  float checkData[] = {
    1, 0, 0,
    0, 1, 0,
    0, 0, 1
  };

  torch::Tensor check = torch::from_blob(checkData, {3, 3});

  assert(tensor.allclose(check));
}