Package ai.djl.pytorch.engine

Class PtNDArrayIndexer

java.lang.Object
ai.djl.ndarray.index.NDArrayIndexer
ai.djl.pytorch.engine.PtNDArrayIndexer
public class PtNDArrayIndexer extends ai.djl.ndarray.index.NDArrayIndexer
The NDArrayIndexer used by the PtNDArray.

  • 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)
    ai.djl.ndarray.NDArray
    get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index)
    void
    set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.dim.NDIndexBooleans indices, ai.djl.ndarray.NDArray value)
    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)
    void
    set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index, Object data)

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

    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

    • get

      public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index)
      Overrides:
      get in class ai.djl.ndarray.index.NDArrayIndexer

    • set

      public void set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index, Object data)
      Overrides:
      set 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.dim.NDIndexBooleans indices, ai.djl.ndarray.NDArray value)
      Overrides:
      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