package torch

  1. Overview
  2. Docs
type t
val create : ?resize:(int * int) -> dir:string -> unit -> t
val random_batch : t -> batch_size:int -> Torch.Tensor.t
val reset : t -> unit
val next_batch : t -> batch_size:int -> Torch.Tensor.t option
OCaml

Innovation. Community. Security.