LaurentMazare / ocaml-torch

OCaml bindings for PyTorch
Apache License 2.0
412 stars 38 forks source link

Update for libtorch 1.9 #57

Closed LaurentMazare closed 3 years ago

LaurentMazare commented 3 years ago

Similar to https://github.com/LaurentMazare/tch-rs/pull/378 this also changes the default naming scheme for the generated functions. Rather than numbered function, the overload_name is used, e.g. Tensor.squeeze1 is renamed to Tenosr.squeeze_dim.