package kaun

You can search for identifiers within the package.

in-package search v0.2.0

kaun
- Library kaun
  - Kaun
    
    Rngs
    
    Metrics
    
    Dataset
    
    Loss
    
    Initializer
    
    Layer
    
    Optimizer
- Library kaun.datasets
  - Kaun_datasets
- Sources
  - kaun
    
    initializers.ml
    
    kaun.ml
    
    kaun__.ml
  - kaun.datasets
    
    kaun_datasets.ml

Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module `Kaun_datasets`Source

Ready-to-use datasets for Kaun

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

package kaun

Module Kaun_datasetsSource

Module `Kaun_datasets`Source