陷阱:为Pytorch C ++ API(libtorch)进行绑定 源码

高奇 总览 Gotch为Pytorch C ++ API(Libtorch)创建了一个薄包装器,以利用其已优化的C ++张量API(约1400多种)和带有CUDA支持的动态图计算,并提供惯用的Go API以在Go中开发和实现深度学习。 一些功能是 全面的Pytorch张量API(〜1404) 功能齐全的Pytorch动态图计算 JIT界面运行使用PyTorch Python API训练/保存的模型 加载预训练的Pytorch模型并进行推理 Pure Go API通过CPU和GPU支持来构建和训练神经网络模型 最新影像模型 NLP语言模型-使用GoTch和构建的单独软件包中的 。