Package ai.djl.pytorch.jni
Class IValue
- java.lang.Object
-
- ai.djl.util.NativeResource<java.lang.Long>
-
- ai.djl.pytorch.jni.IValue
-
- All Implemented Interfaces:
java.lang.AutoCloseable
public class IValue extends ai.djl.util.NativeResource<java.lang.Long>
A class represent a PyTorchIValue
data.DJL doesn't support creating nested IValue.
-
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description void
close()
static IValue
from(boolean value)
Creates a newIValue
of typeboolean
.static IValue
from(double value)
Creates a newIValue
of typedouble
.static IValue
from(long value)
Creates a newIValue
of typelong
.static IValue
from(PtNDArray array)
Creates a newIValue
of typePtNDArray
.static IValue
from(java.lang.String value)
Creates a newIValue
of typeString
.java.lang.String
getType()
Returns the type of the IValue.boolean
isBoolean()
Returns if the IValue is aboolean
type.boolean
isBooleanList()
Returns if the IValue is aboolean[]
type.boolean
isDouble()
Returns if the IValue is adouble
type.boolean
isDoubleList()
Returns if the IValue is adouble[]
type.boolean
isList()
Returns if the IValue is aIValue[]
type.boolean
isLong()
Returns if the IValue is along
type.boolean
isLongList()
Returns if the IValue is along[]
type.boolean
isMap()
Returns if the IValue is aMap<String, V>
type.boolean
isString()
Returns if the IValue is aString
type.boolean
isTensor()
Returns if the IValue is aTensor
type.boolean
isTensorList()
Returns if the IValue is aIValue[]
type.boolean
isTuple()
Returns if the IValue is a tuple type.static IValue
listFrom(boolean... list)
Creates a newIValue
of typeboolean[]
.static IValue
listFrom(double... list)
Creates a newIValue
of typedouble[]
.static IValue
listFrom(long... list)
Creates a newIValue
of typelong[]
.static IValue
listFrom(PtNDArray... list)
Creates a newIValue
of typeNDArray[]
.static IValue
listFrom(IValue... list)
Creates a newIValue
of typeNDArray[]
.static IValue
stringIValueMapFrom(java.util.Map<java.lang.String,IValue> map)
Creates a newIValue
of typeMap[String, IValue]
.static IValue
stringMapFrom(java.util.Map<java.lang.String,PtNDArray> map)
Creates a newIValue
of typeMap[String, PtNDArray]
.boolean
toBoolean()
Returns theboolean
value of this IValue.boolean[]
toBooleanArray()
Returns theboolean[]
value of this IValue.double
toDouble()
Returns thedouble
value of this IValue.double[]
toDoubleArray()
Returns thedouble[]
value of this IValue.IValue[]
toIValueArray()
Returns theIValue[]
value of this IValue list.java.util.Map<java.lang.String,IValue>
toIValueMap()
Returns theMap<String, IValue>
value of this IValue.IValue[]
toIValueTuple()
Returns theMap<String, IValue>
value of this IValue.long
toLong()
Returns thelong
value of this IValue.long[]
toLongArray()
Returns thelong[]
value of this IValue.ai.djl.ndarray.NDList
toNDList(PtNDManager manager)
Returns theNDList
value of this IValue.java.lang.String
toStringValue()
Returns theString
value of this IValue.PtNDArray
toTensor(PtNDManager manager)
Returns theNDArray
value of this IValue.PtNDArray[]
toTensorArray(PtNDManager manager)
Returns theNDArray[]
value of this IValue.static IValue
tupleFrom(IValue... list)
Creates a newIValue
of typeNDArray[]
.
-
-
-
Method Detail
-
getType
public java.lang.String getType()
Returns the type of the IValue.- Returns:
- the type of the IValue
-
isTensor
public boolean isTensor()
Returns if the IValue is aTensor
type.- Returns:
- if the IValue is a Tensor type
-
isBoolean
public boolean isBoolean()
Returns if the IValue is aboolean
type.- Returns:
- if the IValue is a boolean type
-
isLong
public boolean isLong()
Returns if the IValue is along
type.- Returns:
- if the IValue is a long type
-
isDouble
public boolean isDouble()
Returns if the IValue is adouble
type.- Returns:
- if the IValue is a double type
-
isString
public boolean isString()
Returns if the IValue is aString
type.- Returns:
- if the IValue is a String type
-
isBooleanList
public boolean isBooleanList()
Returns if the IValue is aboolean[]
type.- Returns:
- if the IValue is a boolean[] type
-
isLongList
public boolean isLongList()
Returns if the IValue is along[]
type.- Returns:
- if the IValue is a long[] type
-
isDoubleList
public boolean isDoubleList()
Returns if the IValue is adouble[]
type.- Returns:
- if the IValue is a double[] type
-
isTensorList
public boolean isTensorList()
Returns if the IValue is aIValue[]
type.The elements in the array must have the same type.
- Returns:
- if the IValue is a IValue[] type
-
isList
public boolean isList()
Returns if the IValue is aIValue[]
type.The elements in the array must have the same type.
- Returns:
- if the IValue is a IValue[] type
-
isMap
public boolean isMap()
Returns if the IValue is aMap<String, V>
type.- Returns:
- if the IValue is a Map<String, V> type
-
isTuple
public boolean isTuple()
Returns if the IValue is a tuple type.- Returns:
- if the IValue is a tuple type
-
from
public static IValue from(PtNDArray array)
Creates a newIValue
of typePtNDArray
.- Parameters:
array
- the NDArray- Returns:
- a new
IValue
of typePtNDArray
-
from
public static IValue from(boolean value)
Creates a newIValue
of typeboolean
.- Parameters:
value
- the boolean value- Returns:
- a new
IValue
of typeboolean
-
from
public static IValue from(long value)
Creates a newIValue
of typelong
.- Parameters:
value
- the long value- Returns:
- a new
IValue
of typelong
-
from
public static IValue from(double value)
Creates a newIValue
of typedouble
.- Parameters:
value
- the double value- Returns:
- a new
IValue
of typedouble
-
from
public static IValue from(java.lang.String value)
Creates a newIValue
of typeString
.- Parameters:
value
- the String value- Returns:
- a new
IValue
of typeString
-
listFrom
public static IValue listFrom(boolean... list)
Creates a newIValue
of typeboolean[]
.- Parameters:
list
- the boolean[] value- Returns:
- a new
IValue
of typeboolean[]
-
listFrom
public static IValue listFrom(long... list)
Creates a newIValue
of typelong[]
.- Parameters:
list
- the long[] value- Returns:
- a new
IValue
of typelong[]
-
listFrom
public static IValue listFrom(double... list)
Creates a newIValue
of typedouble[]
.- Parameters:
list
- the double[] value- Returns:
- a new
IValue
of typedouble[]
-
listFrom
public static IValue listFrom(PtNDArray... list)
Creates a newIValue
of typeNDArray[]
.- Parameters:
list
- the NDArray[] value- Returns:
- a new
IValue
of typeNDArray[]
-
listFrom
public static IValue listFrom(IValue... list)
Creates a newIValue
of typeNDArray[]
.- Parameters:
list
- the NDArray[] value- Returns:
- a new
IValue
of typeNDArray[]
-
tupleFrom
public static IValue tupleFrom(IValue... list)
Creates a newIValue
of typeNDArray[]
.- Parameters:
list
- the NDArray[] value- Returns:
- a new
IValue
of typeNDArray[]
-
stringMapFrom
public static IValue stringMapFrom(java.util.Map<java.lang.String,PtNDArray> map)
Creates a newIValue
of typeMap[String, PtNDArray]
.- Parameters:
map
- the Map[String, IValue] value- Returns:
- a new
IValue
of typeMap[String, PtNDArray]
-
stringIValueMapFrom
public static IValue stringIValueMapFrom(java.util.Map<java.lang.String,IValue> map)
Creates a newIValue
of typeMap[String, IValue]
.- Parameters:
map
- the Map[String, IValue] value- Returns:
- a new
IValue
of typeMap[String, IValue]
-
toBoolean
public boolean toBoolean()
Returns theboolean
value of this IValue.- Returns:
- the boolean value of this IValue
-
toLong
public long toLong()
Returns thelong
value of this IValue.- Returns:
- the long value of this IValue
-
toDouble
public double toDouble()
Returns thedouble
value of this IValue.- Returns:
- the double value of this IValue
-
toStringValue
public java.lang.String toStringValue()
Returns theString
value of this IValue.- Returns:
- the String value of this IValue
-
toBooleanArray
public boolean[] toBooleanArray()
Returns theboolean[]
value of this IValue.- Returns:
- the boolean[] value of this IValue
-
toLongArray
public long[] toLongArray()
Returns thelong[]
value of this IValue.- Returns:
- the long[] value of this IValue
-
toDoubleArray
public double[] toDoubleArray()
Returns thedouble[]
value of this IValue.- Returns:
- the double[] value of this IValue
-
toTensor
public PtNDArray toTensor(PtNDManager manager)
Returns theNDArray
value of this IValue.- Parameters:
manager
- theNDManager
to create the NDArray- Returns:
- the NDArray value of this IValue
-
toTensorArray
public PtNDArray[] toTensorArray(PtNDManager manager)
Returns theNDArray[]
value of this IValue.- Parameters:
manager
- the NDManager to create NDArray- Returns:
- the NDArray[] value of this IValue
-
toIValueArray
public IValue[] toIValueArray()
Returns theIValue[]
value of this IValue list.- Returns:
- the IValue[] value of this IValue list
-
toIValueMap
public java.util.Map<java.lang.String,IValue> toIValueMap()
Returns theMap<String, IValue>
value of this IValue.- Returns:
- the Map<String, IValue> value of this IValue
-
toIValueTuple
public IValue[] toIValueTuple()
Returns theMap<String, IValue>
value of this IValue.- Returns:
- the Map<String, IValue> value of this IValue
-
toNDList
public ai.djl.ndarray.NDList toNDList(PtNDManager manager)
Returns theNDList
value of this IValue.- Parameters:
manager
- the NDManager to create NDArray- Returns:
- the
NDList
value of this IValue
-
close
public void close()
- Specified by:
close
in interfacejava.lang.AutoCloseable
- Overrides:
close
in classai.djl.util.NativeResource<java.lang.Long>
-
-