package kaun

You can search for identifiers within the package.

in-package search v0.2.0

Flax-inspired neural network library for OCaml

Install

Dune Dependency

github.com Readme Changelog Edit opam file Versions (1)

Authors

Thibaut Mattio

Maintainers

Thibaut Mattio

Sources

raven-1.0.0.alpha0.tbz

sha256=a9a8a9787f8250337187bb7b21cb317c41bfd2ecf08bcfe0ab407c7b6660764d

sha512=fe13cf257c487e41efe2967be147d80fa94bac8996d3aab2b8fd16f0bbbd108c15e0e58c025ec9bf294d4a0d220ca2ba00c3b1b42fa2143f758c5f0ee4c15782

doc/kaun.datasets/Kaun_datasets/index.html

Module `Kaun_datasets`Source

Ready-to-use datasets for 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