We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Int
1 parent 6e3c23a commit fa580ddCopy full SHA for fa580dd
1 file changed
src/ArrayFire/Index.hs
@@ -41,7 +41,14 @@ index (Array fptr) seqs =
41
n = fromIntegral (length seqs)
42
43
-- | Lookup an Array by keys along a specified dimension
44
-lookup :: Array a -> Array a -> Int -> Array a
+lookup
45
+ :: Array a
46
+ -- ^ Input Array
47
+ -> Array Int
48
+ -- ^ Indices
49
+ -> Int
50
+ -- ^ Dimension
51
+ -> Array a
52
lookup a b n = op2 a b $ \p x y -> af_lookup p x y (fromIntegral n)
53
54
-- | A special value representing the entire axis of an 'Array'.
0 commit comments