Package ai.djl.ndarray.index.dim

Class NDIndexTake

java.lang.Object

ai.djl.ndarray.index.dim.NDIndexTake

All Implemented Interfaces:

NDIndexElement

public class NDIndexTake
extends Object
implements NDIndexElement

An NDIndexElement that gets elements by index in the specified axis.

Constructor Summary

Constructors

Constructor	Description
NDIndexTake(NDArray index)	Constructs a pick.

Method Summary

Modifier and Type	Method	Description
NDArray	getIndex()	Returns the index to pick.
int	getRank()	Returns the number of dimensions occupied by this index element.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

NDIndexTake

public NDIndexTake(NDArray index)

Constructs a pick.

Parameters:

index - the index to pick

Method Details

getRank

public int getRank()

Returns the number of dimensions occupied by this index element.

Specified by:

getRank in interface NDIndexElement

Returns:

the number of dimensions occupied by this index element

getIndex

public NDArray getIndex()

Returns the index to pick.

Returns:

the index to pick