[][src]Enum tch::Device

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

A torch device.

Variants

Cpu

The main CPU device.

Cuda(usize)

The main GPU device.

Methods

impl Device[src]

pub fn cuda_if_available() -> Device[src]

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

Trait Implementations

impl Clone for Device[src]

impl Copy for Device[src]

impl Debug for Device[src]

impl Eq for Device[src]

impl PartialEq<Device> for Device[src]

impl StructuralEq for Device[src]

impl StructuralPartialEq for Device[src]

Auto Trait Implementations

impl RefUnwindSafe for Device

impl Send for Device

impl Sync for Device

impl Unpin for Device

impl UnwindSafe for Device

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized
[src]

impl<T> Borrow<T> for T where
    T: ?Sized
[src]

impl<T> BorrowMut<T> for T where
    T: ?Sized
[src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, 
[src]

impl<T> ToOwned for T where
    T: Clone
[src]

type Owned = T

The resulting type after obtaining ownership.

impl<T, U> TryFrom<U> for T where
    U: Into<T>, 
[src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, 
[src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.