文件名称：ocaml-torch:PyTorch的OCaml绑定
文件大小：4.83MB
文件格式：ZIP
更新时间：2024-05-25 01:58:22
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