packages feed

deeplearning-hs-0.1.0.0: Main.hs

{-# LANGUAGE TypeOperators #-}
module Main where

import           Data.Array.Repa
import           DeepLearning.ConvNet
import           DeepLearning.Util

-- |Main
main :: IO ()
main = do
  (pvol, acts) <- withActivations (testNet testShape 2) (testInput testShape)
  print (computeS pvol :: Vol DIM1)
  print acts