Kaun_datasets (u.700e38feb29c357768032d0099b0c576.kaun.1.0.0~alpha0.doc.kaun.datasets.Kaun

Source

val mnist : 
  ?train:bool ->
  ?flatten:bool ->
  ?normalize:bool ->
  ?data_format:[ `NCHW | `NHWC ] ->
  device:'dev Kaun.device ->
  unit ->
  ((Bigarray.float32_elt, 'dev) Kaun.tensor
   * (Bigarray.float32_elt, 'dev) Kaun.tensor)
    Kaun.Dataset.t

MNIST handwritten digits dataset