Torch.Typed.Vision

Documentation

data MNIST (m :: Type -> Type) (device :: (DeviceType, Nat)) (batchSize :: Nat)

Constructors

MNIST

Fields

• mnistData :: MnistData

Instances

(KnownNat batchSize, KnownDevice device, Applicative m) => Dataset (m :: Type -> Type) (MNIST m device batchSize) Int ((Tensor device 'Float '[batchSize, 784], Tensor device 'Int64 '[batchSize]) :: Type)

Defined in Torch.Typed.Vision

Methods

getItem :: MNIST m device batchSize -> Int -> m (Tensor device 'Float '[batchSize, 784], Tensor device 'Int64 '[batchSize])

keys :: MNIST m device batchSize -> Set Int

data MnistData

Constructors

MnistData

Fields

• images :: ByteString
• labels :: ByteString

type Rows = 28

type Cols = 28

type DataDim = Rows * Cols

type ClassDim = 10

getLabels :: forall (n :: Nat). KnownNat n => MnistData -> [Int] -> CPUTensor 'Int64 '[n]

getLabel :: MnistData -> Int -> Int

getImage :: MnistData -> Int -> CPUTensor 'Float '[DataDim]

getImages' :: forall (n :: Nat). KnownNat n => MnistData -> [Int] -> CPUTensor 'Float '[n, DataDim]

getImages :: forall (n :: Nat). KnownNat n => MnistData -> [Int] -> CPUTensor 'Float '[n, DataDim]

length :: MnistData -> Int

decompressFile :: String -> String -> IO ByteString

initMnist :: String -> IO (MnistData, MnistData)