Skip to content

Libtorch notes

Jump to bottom
Saurav Agarwal edited this page Oct 2, 2023 · 1 revision

CPP extensions

Importing tensors from CPP to Python

Example of C++ and Python

export Torch_DIR=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'`
Clone this wiki locally