【文件属性】:
文件名称:ocaml-torch:PyTorch的OCaml绑定
文件大小:4.83MB
文件格式:ZIP
更新时间:2021-05-05 08:11:42
machine-learning deep-learning neural-network gpu ocaml
火炬
ocaml-torch为张量库提供了一些ocaml绑定。 借助GPU加速和基于磁带的自动微分功能,将OCaml NumPy样张量计算引入。
这些绑定使用 ,并且大多数是自动生成的。 当前的GitHub技巧和opam软件包v0.7对应于PyTorch v1.8.0 。
在Linux上请注意,您将需要使用cxx11 abi , 的PyTorch 。
Opam安装
可以使用以下命令安装软件包。 这将自动安装CPU版本的libtorch。
opam install torch
然后,您可以编译一些示例代码,请参阅下面的一些说明。 ocaml-torch也可以通过或在交互模式下使用。
这是一个示例utop会话。
建立一个简单的例子
要构建第一个割炬程序,请创建包含以下内容的文件example.ml 。
open Torch
let () =
let tensor = Tenso