ocaml-torch:PyTorch的OCaml绑定

时间:2021-05-05 08:11:42
【文件属性】:
文件名称: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

网友评论