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