文件名称:go-torch:Golang的LibTorch(PyTorch)绑定
文件大小:16KB
文件格式:ZIP
更新时间:2024-06-02 19:45:09
golang machine-learning pytorch golang-library Go
一切都真的过时了。 对于最新的绑定,我建议您退出或其他可用软件包。 进展中的工作...需自担风险:-) 火炬 Golang的LibTorch(PyTorch)绑定。 库的首要目的是针对从Python版本的PyTorch导出的序列化模型进行推理。 库也可以用于直接从Go编译TorchScript应用程序。 正在安装 $ go get github.com/orktes/go-torch 用法 go-torch取决于LibTorch共享库的可用性。 有关更多信息,请参阅 。 这也是用于执行库测试的示例 。 import ( "github.com/orktes/go-torch" ) 创建张量 支持的标量类型: 火炬字节uint8 火炬char int8 torch.Int int32 火炬Long int64 火炬浮动float32 火炬。双float64 matri
【文件预览】:
go-torch-master
----.travis.yml(60B)
----lib.go(575B)
----jit.go(7KB)
----tensor.go(8KB)
----example_test.go(402B)
----errors.go(423B)
----LICENSE(727B)
----TODO.md(232B)
----torch.hpp(2KB)
----jit_test.go(5KB)
----.gitignore(199B)
----dtype.go(961B)
----Makefile(131B)
----tuple.go(633B)
----tensor_test.go(974B)
----README.md(3KB)
----torch.cpp(11KB)
----scripts()
--------Dockerfile(873B)
--------docker-compose.yml(510B)