Package ai.djl.engine.rust

Class RsNDArrayIndexer

java.lang.Object
ai.djl.ndarray.index.NDArrayIndexer
ai.djl.engine.rust.RsNDArrayIndexer
public class RsNDArrayIndexer extends ai.djl.ndarray.index.NDArrayIndexer
The NDArrayIndexer used by the RsNDArray.

  • Method Summary

    Modifier and Type
    Method
    Description
    ai.djl.ndarray.NDArray
    get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullPick fullPick)
    ai.djl.ndarray.NDArray
    get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice)
    ai.djl.ndarray.NDArray
    get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullTake fullTake)
    void
    set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, ai.djl.ndarray.NDArray value)
    void
    set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, Number value)

    Methods inherited from class ai.djl.ndarray.index.NDArrayIndexer

    get, set, set, setScalar

    Methods inherited from class java.lang.Object

    clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

  • Method Details

    • get

      public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullPick fullPick)
      Specified by:
      get in class ai.djl.ndarray.index.NDArrayIndexer

    • get

      public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullTake fullTake)
      Specified by:
      get in class ai.djl.ndarray.index.NDArrayIndexer

    • get

      public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice)
      Specified by:
      get in class ai.djl.ndarray.index.NDArrayIndexer

    • set

      public void set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, ai.djl.ndarray.NDArray value)
      Specified by:
      set in class ai.djl.ndarray.index.NDArrayIndexer

    • set

      public void set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, Number value)
      Specified by:
      set in class ai.djl.ndarray.index.NDArrayIndexer