Torch.FFI.TestsNN

Documentation

type NNNum r = (Fractional r, Show r, Eq r)

data NNTestSuite state ten real accreal gen

Constructors

NNTestSuite

Fields

• _newWithSize1d :: state -> CLLong -> IO ten
• _newWithSize2d :: state -> CLLong -> CLLong -> IO ten
• _newGen :: IO gen
• _normal :: Either (state -> ten -> gen -> CDouble -> CDouble -> IO ()) (state -> ten -> CDouble -> CDouble -> IO ())
• _fill :: state -> ten -> real -> IO ()
• _sumall :: state -> ten -> IO accreal
• _free :: state -> ten -> IO ()
• _nnAbsUpdateOutput :: state -> ten -> ten -> IO ()
• _nnHSUpdateOutput :: Maybe (state -> ten -> ten -> CDouble -> IO ())
• _nnL1UpdateOutput :: state -> ten -> ten -> IO ()
• _nnRReLUUpdateOutput :: state -> ten -> ten -> ten -> CDouble -> CDouble -> CBool -> CBool -> gen -> IO ()

testSuite :: (NNNum real, NNNum accreal) => state -> NNTestSuite state ten real accreal gen -> Spec