Package ai.djl.ndarray.index.full

Class NDIndexFullTake

java.lang.Object

ai.djl.ndarray.index.full.NDIndexFullTake

public final class NDIndexFullTake
extends Object

A simplified representation of a take-based NDIndex.

Method Summary

Modifier and Type	Method	Description
static Optional<NDIndexFullTake>	fromIndex(NDIndex index, Shape target)	Returns (if possible) the NDIndexFullTake representation of an NDIndex.
int	getAxis()	Returns the axis to take.
NDArray	getIndices()	Returns the indices to take.

Methods inherited from class java.lang.Object

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

Method Details

fromIndex

public static Optional<NDIndexFullTake> fromIndex(NDIndex index,
 Shape target)

Returns (if possible) the NDIndexFullTake representation of an NDIndex.

Parameters:

index - the index to represent

target - the shape of the array to index

Returns:

the full take representation or nothing if it can't represent the index

getIndices

public NDArray getIndices()

Returns the indices to take.

Returns:

the indices to take

getAxis

public int getAxis()

Returns the axis to take.

Returns:

the axis to take