package torch

  1. Overview
  2. Docs
Torch bindings for OCaml

Install

Dune Dependency

Authors

Maintainers

Sources

torch-v0.16.0.tar.gz
sha256=ccd9ef3b630bdc7c41e363e71d8ecb86c316460cbf79afe67546c6ff22c19da4

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
OCaml

Innovation. Community. Security.