ai.djl.pytorch.engine

Class PtDeviceType

java.lang.Object

ai.djl.pytorch.engine.PtDeviceType

All Implemented Interfaces:

ai.djl.DeviceType

public final class PtDeviceType
extends java.lang.Object
implements ai.djl.DeviceType

DeviceType is the PyTorch equivalent of the types in Device.

Method Summary

Modifier and Type	Method and Description
static java.lang.String	fromDeviceType(int deviceType) Converts from an PyTorch device number to Device.
static int	toDeviceType(ai.djl.Device device) Converts a Device to the corresponding PyTorch device number.

Methods inherited from class java.lang.Object

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

Method Detail

toDeviceType

public static int toDeviceType(ai.djl.Device device)

Converts a Device to the corresponding PyTorch device number.

Specified by:

toDeviceType in interface ai.djl.DeviceType

Parameters:

device - the java Device

Returns:

the PyTorch device number

fromDeviceType

public static java.lang.String fromDeviceType(int deviceType)

Converts from an PyTorch device number to Device.

Specified by:

fromDeviceType in interface ai.djl.DeviceType

Parameters:

deviceType - the PyTorch device number

Returns:

the corresponding Device