package torch

  1. Overview
  2. Docs
PyTorch bindings for OCaml

Install

Dune Dependency

Authors

Maintainers

Sources

0.12.tar.gz
md5=ed7f1c706ae1d8f70c0db4c6cde1d391
sha512=cc09079eb8b1c876f7416b6446c9f5439b7fb54e82d8cbad778ef8bd59d62dc2681e84f54fc69c974a0ed67d3a435e6108b4303cc1c424b8dc89fdfa5992d9d7

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
OCaml

Innovation. Community. Security.