tch rs:PyTorch的C ++ API的Rust绑定 源码
频道 PyTorch的C ++ API的Rust绑定。 tch板条箱的目标是在C ++ PyTorch api(又名libtorch)周围提供一些薄包装纸。 它旨在保持与原始C ++ API的距离。 然后可以在此基础上开发更多惯用的防锈绑定。 该可以在docs.rs上找到。 libtorch之上的C api的代码生成部分来自 。 入门 这个箱子需要在版本V1.7.0的C ++ PyTorch库(libtorch)可用您的系统上。 您可以手动安装它,并通过LIBTORCH环境变量让构建脚本知道它。 如果未设置,则构建脚本将尝试下载并提取libtorch的预构建二进制版本。 Libtorch手动
文件列表
tch-rs:PyTorch的C ++ API的Rust绑定
(预估有个146文件)
.gitignore
87B
LICENSE
3KB
Makefile
229B
config
120B
stb_image.h
256KB
stb_image_resize.h
113KB
torch_api.cpp
34KB
dummy_cuda_dependency.cpp
177B
torch_api_generated.cpp.h
371KB
stb_image_write.h
65KB
暂无评论