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.5.tar.gz
md5=112bc4eef842883abdf25e150dc7952b
sha512=1670f337e5a00aa6acda9182b71c586cd817ed565ab1df280bba30f610760affb0d1fc859abe5f33bf78323cf3bf2391ad0b3d955aedff3894aae70723997705

doc/torch.vision/Torch_vision/Image/Loader/index.html

Module Image.LoaderSource

Sourcetype t
Sourceval create : ?resize:(Base.int * Base.int) -> dir:Base.string -> Base.unit -> t
Sourceval random_batch : t -> batch_size:Base.int -> Torch.Tensor.t
Sourceval reset : t -> Base.unit
Sourceval next_batch : t -> batch_size:Base.int -> Torch.Tensor.t Base.option