enum

ML::Tensor::Device

Inherits Enum / Comparable / Value / Object

Device location

Constants

CPU = 0
GPU = 1

Instance methods

cpu?

Returns true if this enum value equals CPU

Source
gpu?

Returns true if this enum value equals GPU

Source