ai.djl.ndarray.index

Class NDIndexFullSlice

java.lang.Object

ai.djl.ndarray.index.NDIndexFullSlice

public class NDIndexFullSlice
extends java.lang.Object

An index as a slice on all dimensions where some dimensions can be squeezed.

Create using NDIndex.getAsFullSlice(Shape).

Method Summary

Modifier and Type	Method and Description
long[]	getMax() Returns the slice max for each axis.
long[]	getMin() Returns the slice min for each axis.
Shape	getShape() Returns the slice shape without squeezing.
Shape	getSqueezedShape() Returns the slice shape with squeezing.
long[]	getStep() Returns the slice step for each axis.
java.util.List<java.lang.Integer>	getToSqueeze() Returns the list of axis to squeeze.

Methods inherited from class java.lang.Object

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

Method Detail

getMin

public long[] getMin()

Returns the slice min for each axis.

Returns:

the slice min for each axis

getMax

public long[] getMax()

Returns the slice max for each axis.

Returns:

the slice max for each axis

getStep

public long[] getStep()

Returns the slice step for each axis.

Returns:

the slice step for each axis

getToSqueeze

public java.util.List<java.lang.Integer> getToSqueeze()

Returns the list of axis to squeeze.

Returns:

the list of axis to squeeze

getShape

public Shape getShape()

Returns the slice shape without squeezing.

Returns:

the slice shape without squeezing

getSqueezedShape

public Shape getSqueezedShape()

Returns the slice shape with squeezing.

Returns:

the slice shape with squeezing