Copyright | (c) Sam Stites 2017 |
---|---|
License | BSD3 |
Maintainer | sam@stites.io |
Stability | experimental |
Portability | non-portable |
Safe Haskell | None |
Language | Haskell2010 |
Synopsis
- _gather :: Tensor d -> Tensor d -> Word -> IndexTensor '[n] -> IO ()
- _scatter :: Tensor d -> Word -> IndexTensor '[n] -> Tensor d -> IO ()
- _scatterAdd :: Tensor d -> Word -> IndexTensor '[n] -> Tensor d -> IO ()
- _scatterFill :: Tensor d -> Word -> IndexTensor '[n] -> HsReal -> IO ()
Documentation
_scatter :: Tensor d -> Word -> IndexTensor '[n] -> Tensor d -> IO () Source #
Static call to _scatter
_scatterAdd :: Tensor d -> Word -> IndexTensor '[n] -> Tensor d -> IO () Source #
Static call to _scatterAdd
_scatterFill :: Tensor d -> Word -> IndexTensor '[n] -> HsReal -> IO () Source #
Static call to _scatterFill