package torch

  1. Overview
  2. Docs
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