| 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.