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.15.tar.gz
md5=e0e2cf2688c69d6bb60f837e812db432
sha512=a518928c5be656a85973db2fe2fb8efb73ec4f4f972f0f733cd75ee8c25292d907bc31ca04a3f2dd5bc95800ee15ee19533ca47f9ff43432485c528c8f08ca90

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

Module Imagenet.LoaderSource

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