Documentation

Index into an Array by Seq

Lookup an Array by keys along a specified dimension

Assign values into an Array range defined by Seq indices

>>> let a = vector @Double 5 [1..]
>>> assignSeq a [Seq 1 3 1] (vector @Double 3 [0,0,0])

Index into an Array using generalized Index values (arrays or sequences)

>>> let a = matrix @Double (3,3) [[1..],[1..],[1..]]
>>> indexGen a [seqIdx (Seq 0 1 1) False, seqIdx (Seq 0 1 1) False]

Assign values into an Array using generalized Index values

>>> let a = matrix @Double (3,3) [[1..],[1..],[1..]]
>>> let b = matrix @Double (2,2) [[0,0],[0,0]]
>>> assignGen a [seqIdx (Seq 0 1 1) False, seqIdx (Seq 0 1 1) False] b

A special Seq value representing the entire axis of an Array. Hard-coded from include/af/seq.h because FFI cannot import static const values.

Select the full extent of a dimension. Use in tuple indices where you want all elements along an axis.

arr ! (range 0 2, full, at 1)

Convert index expressions to a list of Index. Supports a single Index or tuples of up to four Index values (matching ArrayFire's maximum of 4 dimensions).

Lift a Seq to an Index for use in tuple-based indexing.

Index an Array. Accepts a single Index or a tuple of up to four.

arr ! at 0                      -- 1D: element 0
arr ! range 1 3                 -- 1D: rows 1-3
arr ! (range 0 2, at 1)         -- 2D
arr ! (range 0 2, full, at 1)   -- 3D, full second axis

Assign into a range of an Array. Lens-style: use with (&).

arr & range 1 3 .~ src
arr & (range 0 1, at 2) .~ src