Shortcuts

DType

public enum DType

Codes representing tensor data types.

Enum Constants

FLOAT32

public static final DType FLOAT32

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

FLOAT64

public static final DType FLOAT64

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

INT32

public static final DType INT32

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

INT64

public static final DType INT64

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

INT8

public static final DType INT8

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

UINT8

public static final DType UINT8

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

Fields

jniCode

final int jniCode

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources