package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type
type t
val create : ?resize:(Base.int * Base.int) -> dir:Base.string -> Base.unit -> t
val random_batch : t -> batch_size:Base.int -> Torch.Tensor.t
val reset : t -> Base.unit
val next_batch : t -> batch_size:Base.int -> Torch.Tensor.t Base.option