| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
Torch.FFI.THC.Int.TensorTopK
Synopsis
- c_topk :: Ptr C'THCState -> Ptr C'THCudaIntTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> CLLong -> CInt -> CInt -> CInt -> IO ()
- p_topk :: FunPtr (Ptr C'THCState -> Ptr C'THCudaIntTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> CLLong -> CInt -> CInt -> CInt -> IO ())
Documentation
c_topk :: Ptr C'THCState -> Ptr C'THCudaIntTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> CLLong -> CInt -> CInt -> CInt -> IO () Source #
c_topk : state topK indices input k dim dir sorted -> void
p_topk :: FunPtr (Ptr C'THCState -> Ptr C'THCudaIntTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> CLLong -> CInt -> CInt -> CInt -> IO ()) Source #
p_topk : Pointer to function : state topK indices input k dim dir sorted -> void