org.pytorch

Enum DType

java.lang.Object

java.lang.Enum<DType>

org.pytorch.DType

All Implemented Interfaces:

java.io.Serializable, java.lang.Comparable<DType>

public enum DType
extends java.lang.Enum<DType>

Codes representing tensor data types.

Enum Constant Summary

Enum Constants

Enum Constant and Description
FLOAT32 Code for dtype torch.float32.
FLOAT64 Code for dtype torch.float64.
INT32 Code for dtype torch.int32.
INT64 Code for dtype torch.int64.
INT8 Code for dtype torch.int8.
UINT8 Code for dtype torch.uint8.

Method Summary

Modifier and Type	Method and Description
static DType	valueOf(java.lang.String name) Returns the enum constant of this type with the specified name.
static DType[]	values() Returns an array containing the constants of this enum type, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Detail

UINT8

public static final DType UINT8

Code for dtype torch.uint8. Tensor.dtype()

INT8

public static final DType INT8

Code for dtype torch.int8. Tensor.dtype()

INT32

public static final DType INT32

Code for dtype torch.int32. Tensor.dtype()

FLOAT32

public static final DType FLOAT32

Code for dtype torch.float32. Tensor.dtype()

INT64

public static final DType INT64

Code for dtype torch.int64. Tensor.dtype()

FLOAT64

public static final DType FLOAT64

Code for dtype torch.float64. Tensor.dtype()

Method Detail

values

public static DType[] values()

Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows:

for (DType c : DType.values())
    System.out.println(c);

Returns:

an array containing the constants of this enum type, in the order they are declared

valueOf

public static DType valueOf(java.lang.String name)

Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

java.lang.IllegalArgumentException - if this enum type has no constant with the specified name

java.lang.NullPointerException - if the argument is null