mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-07-14 14:10:33 +03:00
8 lines
137 B
C++
8 lines
137 B
C++
|
#include <torch/torch.h>
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
int main() {
|
||
|
|
torch::Tensor tensor = torch::eye(3);
|
||
|
|
std::cout << tensor << std::endl;
|
||
|
|
}
|