ai.djl.pytorch.jni

Class IValueUtils

java.lang.Object

ai.djl.pytorch.jni.IValueUtils

public final class IValueUtils
extends java.lang.Object

IValueUtils is utility class to deal with IValue in PyTorch.

Method Summary

Modifier and Type	Method and Description
static ai.djl.ndarray.NDList	forward(PtSymbolBlock block, ai.djl.ndarray.NDList inputs) Run the forward of PyTorch module.
static boolean	isArray(Pointer iValueHandle) Check IValue is a container of IValue Array.
static boolean	isMap(Pointer iValueHandle) Check IValue is a container of IValue Map.
static boolean	isNDArray(Pointer iValueHandle) Check IValue is a container of PtNDArray.
static boolean	isNDList(Pointer iValueHandle) Check IValue is a container of NDList.
static boolean	isString(Pointer iValueHandle) Check IValue is a container of String.
static boolean	isTuple(Pointer iValueHandle) Check IValue is a container of IValue Tuple.
static Pointer[]	toIValueArray(Pointer iValueHandle) Extract IValue to an IValue Array.
static java.util.Map<Pointer,Pointer>	toIValueMap(Pointer iValueHandle) Extract IValue to a Map.
static Pointer	toIValuePointer(PtNDArray array) Create IValue Pointer from NDArray.
static PtNDArray	toNDArray(Pointer iValueHandle, PtNDManager manager) Extract IValue with a PtNDArray value.
static ai.djl.ndarray.NDList	toNDList(Pointer iValueHandle, PtNDManager manager) Extract IValue to NDList.
static java.lang.String	toString(Pointer iValueHandle) Extract IValue to String.

Methods inherited from class java.lang.Object

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

Method Detail

toIValuePointer

public static Pointer toIValuePointer(PtNDArray array)

Create IValue Pointer from NDArray.

Parameters:

array - PtNDArray

Returns:

IValue Pointer

isNDArray

public static boolean isNDArray(Pointer iValueHandle)

Check IValue is a container of PtNDArray.

Parameters:

iValueHandle - IValue Pointer

Returns:

result

isNDList

public static boolean isNDList(Pointer iValueHandle)

Check IValue is a container of NDList.

Parameters:

iValueHandle - IValue Pointer

Returns:

result

isArray

public static boolean isArray(Pointer iValueHandle)

Check IValue is a container of IValue Array.

Parameters:

iValueHandle - IValue Pointer

Returns:

result

isTuple

public static boolean isTuple(Pointer iValueHandle)

Check IValue is a container of IValue Tuple.

Parameters:

iValueHandle - IValue Pointer

Returns:

result

isMap

public static boolean isMap(Pointer iValueHandle)

Check IValue is a container of IValue Map.

Parameters:

iValueHandle - IValue Pointer

Returns:

result

isString

public static boolean isString(Pointer iValueHandle)

Check IValue is a container of String.

Parameters:

iValueHandle - IValue Pointer

Returns:

result

toNDArray

public static PtNDArray toNDArray(Pointer iValueHandle,
                                  PtNDManager manager)

Extract IValue with a PtNDArray value.

Parameters:

iValueHandle - IValue Pointer

manager - PtNDManager that creates PtNDArray

Returns:

NDArray

toNDList

public static ai.djl.ndarray.NDList toNDList(Pointer iValueHandle,
                                             PtNDManager manager)

Extract IValue to NDList.

Parameters:

iValueHandle - IValue Pointer

manager - PtNDManager that creates PtNDArray

Returns:

NDList

toString

public static java.lang.String toString(Pointer iValueHandle)

Extract IValue to String.

Parameters:

iValueHandle - IValue Pointer

Returns:

String

toIValueArray

public static Pointer[] toIValueArray(Pointer iValueHandle)

Extract IValue to an IValue Array.

Parameters:

iValueHandle - IValue Pointer

Returns:

IValue array

toIValueMap

public static java.util.Map<Pointer,Pointer> toIValueMap(Pointer iValueHandle)

Extract IValue to a Map.

Parameters:

iValueHandle - IValue Pointer

Returns:

IValue Map

forward

public static ai.djl.ndarray.NDList forward(PtSymbolBlock block,
                                            ai.djl.ndarray.NDList inputs)

Run the forward of PyTorch module.

Parameters:

block - the block that contains PyTorch module.

inputs - input NDList

Returns:

result NDList