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
TheNDArrayIndexer
used by thePtNDArray
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods 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, java.lang.Number value)
void
set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index, java.lang.Object data)
-
-
-
Method Detail
-
get
public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullPick fullPick)
- Specified by:
get
in classai.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 classai.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 classai.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 classai.djl.ndarray.index.NDArrayIndexer
-
set
public void set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index, java.lang.Object data)
- Overrides:
set
in classai.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 classai.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 classai.djl.ndarray.index.NDArrayIndexer
-
set
public void set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, java.lang.Number value)
- Specified by:
set
in classai.djl.ndarray.index.NDArrayIndexer
-
-