package kaun

  1. Overview
  2. Docs

Module Kaun.DatasetSource

Sourcetype 'a t
Sourceval of_xy : (('l1, 'dev) tensor * ('l2, 'dev) tensor) -> (('l1, 'dev) tensor * ('l2, 'dev) tensor) t
Sourceval map : ('a -> 'b) -> 'a t -> 'b t
Sourceval batch : int -> 'a t -> 'a t
Sourceval batch_xy : int -> (('l1, 'dev) tensor * ('l2, 'dev) tensor) t -> (('l1, 'dev) tensor * ('l2, 'dev) tensor) t
Sourceval shuffle : ?seed:int -> 'a t -> 'a t
Sourceval iter : ('a -> unit) -> 'a t -> unit
Sourceval length : 'a t -> int
Sourceval take : int -> 'a t -> 'a list
OCaml

Innovation. Community. Security.