Enum tch::Device

pub enum Device {
    Cpu,
    Cuda(usize),
    Mps,
    Vulkan,
}

A torch device.

Variants

Cpu

The main CPU device.

Cuda(usize)

The main GPU device.

Mps

The main MPS device.

Vulkan

The main Vulkan device.

Implementations

impl Device

pub fn cuda_if_available() -> Device

Returns a GPU device if available, else default to CPU.

pub fn is_cuda(self) -> bool

Trait Implementations

impl Clone for Device

fn clone(&self) -> Device

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Device

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Hash for Device

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq<Device> for Device

fn eq(&self, other: &Device) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for Device

impl Eq for Device

impl StructuralEq for Device

impl StructuralPartialEq for Device