Copyright | (c) Sam Stites 2017 |
---|---|
License | BSD3 |
Maintainer | sam@stites.io |
Stability | experimental |
Portability | non-portable |
Safe Haskell | None |
Language | Haskell2010 |
Torch.Indef.Dynamic.Tensor.Mode
Description
Documentation
Arguments
:: (Dynamic, IndexDynamic) | |
-> Dynamic | |
-> Word | dimension to operate over |
-> Maybe KeepDim | |
-> IO () |
Mutate the tuple of a tensor and index tensor to contain the most frequent element in the specified dimension.
FIXME: if KeepDim is False, we need to return Nothing for the index tensor -- otherwise bad things may happen.