package torch

  1. Overview
  2. Docs
PyTorch bindings for OCaml

Install 

dune-project
 Dependency

github.com Readme Edit opam file Versions (18)

Authors

  1. L
    Laurent Mazare

Maintainers

  1. L
    Laurent Mazare <lmazare@gmail.com>

Sources

0.16.tar.gz
md5=8746b973b125b378d4f5f44800d1a0a6
sha512=86887d16a40f366e16a95f38863c4afd08ceb426b7a7341802451210739f5213322f0d10bb2f979adb26296c4fbee699a3ab5a1d264334f20c44bc05d6f8540e

doc/src/torch.toplevel/torch_toplevel.ml.html

Source file torch_toplevel.ml

1
2
3
4
5
6
7
8
9
let parse_and_execute str =
  Lexing.from_string str
  |> !Toploop.parse_toplevel_phrase
  |> Toploop.execute_phrase true Format.err_formatter
  |> fun (_ : bool) -> ()

let register_pp str = Printf.sprintf "#install_printer %s;;" str |> parse_and_execute
let register_all_pps () = register_pp "Torch.Tensor.pp"
let () = register_all_pps ()