Safe Haskell	None
Language	Haskell2010

Torch.Extensions

Description

Hyper Parameters of OpNet

Synopsis

Documentation

gpu :: Device Source #

GPU

cpu :: Device Source #

CPU

pow10 :: Tensor -> Tensor Source #

The inverse of log10

nanToNum :: Float -> Float -> Float -> Tensor -> Tensor Source #

Because snake_case sucks and this project uses Float instead of Double

nanToNum' :: Tensor -> Tensor Source #

Default limits for nanToNum

nanToNum'' :: Tensor -> Tensor Source #

Default limits for nanToNum (0.0)

mseLoss' :: Reduction -> Tensor -> Tensor -> Tensor Source #

MSE with reduction

meanAll :: Tensor -> Tensor Source #

Mean over all dimensions