Package ai.djl.tensorflow.engine
Class TfNDArrayIndexer
java.lang.Object
ai.djl.ndarray.index.NDArrayIndexer
ai.djl.tensorflow.engine.TfNDArrayIndexer
public class TfNDArrayIndexer
extends ai.djl.ndarray.index.NDArrayIndexer
The
NDArrayIndexer used by the TfNDArray.-
Method Summary
Modifier and TypeMethodDescriptionai.djl.ndarray.NDArrayget(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullPick fullPick) ai.djl.ndarray.NDArrayget(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice) ai.djl.ndarray.NDArrayget(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullTake fullTake) voidset(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, ai.djl.ndarray.NDArray value) voidMethods inherited from class ai.djl.ndarray.index.NDArrayIndexer
get, set, set, setScalar
-
Method Details
-
get
public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullPick fullPick) - Specified by:
getin 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:
getin 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:
getin 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:
setin classai.djl.ndarray.index.NDArrayIndexer
-
set
public void set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.full.NDIndexFullSlice fullSlice, Number value) - Specified by:
setin classai.djl.ndarray.index.NDArrayIndexer
-