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 and 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)
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)

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 Detail

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.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.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,
                java.lang.Number value)

Specified by:

set in class ai.djl.ndarray.index.NDArrayIndexer