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, java.lang.Number value)
void	set(ai.djl.ndarray.NDArray array, ai.djl.ndarray.index.NDIndex index, java.lang.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 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.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,
                java.lang.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,
                java.lang.Number value)

Specified by:

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