horde-ad-0.2.0.0: Higher Order Reverse Derivatives Efficiently - Automatic Differentiation
Safe HaskellNone
LanguageGHC2024

MnistData

Description

Parsing and pre-processing of MNIST data.

Documentation

type LengthTestData = 10000 Source #

type MnistData r = (Vector r, Vector r) Source #

type MnistDataLinearR r = (Ranked 1 r, Ranked 1 r) Source #

type MnistDataR r = (Ranked 2 r, Ranked 1 r) Source #

type MnistDataBatchR r = (Ranked 3 r, Ranked 2 r) Source #

type MnistDataBatchS (batch_size :: Nat) r = (Shaped '[batch_size, SizeMnistHeight, SizeMnistWidth] r, Shaped '[batch_size, SizeMnistLabel] r) Source #

mkMnistDataBatchS :: forall (batch_size :: Nat) r. (Elt r, KnownNat batch_size) => [MnistDataS r] -> MnistDataBatchS batch_size r Source #

shuffle :: StdGen -> [a] -> [a] Source #

chunksOf :: Int -> [e] -> [[e]] Source #