package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type
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