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, boolean isTrain) Run the forward of PyTorch module.
static boolean	isList(long iValueHandle) Check IValue is a container of IValue List.
static boolean	isMap(long iValueHandle) Check IValue is a container of IValue Map.
static boolean	isNDArray(long iValueHandle) Check IValue is a container of PtNDArray.
static boolean	isNDList(long iValueHandle) Check IValue is a container of NDList.
static boolean	isString(long iValueHandle) Check IValue is a container of String.
static boolean	isTuple(long iValueHandle) Check IValue is a container of IValue Tuple.
static long	iValueFromDict(long[] pointers, java.lang.String[] names) Create Dict IValue Pointer from pointer with its key name.
static long	iValueFromList(long[] pointers) Create List IValue Pointer from pointer array.
static long[]	toIValueArray(long iValueHandle) Extract IValue to an IValue Array.
static java.util.Map<java.lang.Long,java.lang.Long>	toIValueMap(long iValueHandle) Extract IValue to a Map.
static long	toIValuePointer(long arrayHandle) Create IValue Pointer from NDArray.
static PtNDArray	toNDArray(long iValueHandle, PtNDManager manager) Extract IValue with a PtNDArray value.
static ai.djl.ndarray.NDList	toNDList(long iValueHandle, PtNDManager manager) Extract IValue to NDList.
static java.lang.String	toString(long 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 long toIValuePointer(long arrayHandle)

Create IValue Pointer from NDArray.

Parameters:

arrayHandle - the handle for PyTorch Tensor

Returns:

IValue Pointer

iValueFromList

public static long iValueFromList(long[] pointers)

Create List IValue Pointer from pointer array.

Parameters:

pointers - pointer array

Returns:

IValue Pointer

iValueFromDict

public static long iValueFromDict(long[] pointers,
                                  java.lang.String[] names)

Create Dict IValue Pointer from pointer with its key name.

Parameters:

pointers - pointer array

names - the key value of the pointer

Returns:

IValue Pointer

isNDArray

public static boolean isNDArray(long iValueHandle)

Check IValue is a container of PtNDArray.

Parameters:

iValueHandle - IValue pointer

Returns:

result

isNDList

public static boolean isNDList(long iValueHandle)

Check IValue is a container of NDList.

Parameters:

iValueHandle - IValue pointer

Returns:

result

isList

public static boolean isList(long iValueHandle)

Check IValue is a container of IValue List.

Parameters:

iValueHandle - IValue pointer

Returns:

result

isTuple

public static boolean isTuple(long iValueHandle)

Check IValue is a container of IValue Tuple.

Parameters:

iValueHandle - IValue pointer

Returns:

result

isMap

public static boolean isMap(long iValueHandle)

Check IValue is a container of IValue Map.

Parameters:

iValueHandle - IValue pointer

Returns:

result

isString

public static boolean isString(long iValueHandle)

Check IValue is a container of String.

Parameters:

iValueHandle - IValue pointer

Returns:

result

toNDArray

public static PtNDArray toNDArray(long 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(long 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(long iValueHandle)

Extract IValue to String.

Parameters:

iValueHandle - IValue pointer

Returns:

String

toIValueArray

public static long[] toIValueArray(long iValueHandle)

Extract IValue to an IValue Array.

Parameters:

iValueHandle - IValue pointer

Returns:

IValue array

toIValueMap

public static java.util.Map<java.lang.Long,java.lang.Long> toIValueMap(long 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,
                                            boolean isTrain)

Run the forward of PyTorch module.

Parameters:

block - the block that contains PyTorch module

inputs - input NDList

isTrain - is running on training mode

Returns:

result NDList