hasktorch-indef-0.0.1.0: Core Hasktorch abstractions wrapping FFI bindings

Copyright(c) Sam Stites 2017
LicenseBSD3
Maintainersam@stites.io
Stabilityexperimental
Portabilitynon-portable
Safe HaskellNone
LanguageHaskell2010

Torch.Indef.Dynamic.Tensor.TopK

Description

 
Synopsis

Documentation

_topk Source #

Arguments

:: (Dynamic, IndexDynamic) 
-> Dynamic 
-> Integer 
-> Word

dimension to operate on

-> TopKOrder 
-> Maybe KeepDim 
-> IO () 

returns all k smallest elements in a tensor over a given dimension, including their indices, in unsorted order.