STenOptions

case class STenOptions(value: TensorOptions)
Companion:
object
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any

Value members

Concrete methods

def cpu[S : Sc]: STenOptions

Returns a copy with device set to CPU

Returns a copy with device set to CPU

def cuda[S : Sc]: STenOptions

Returns a copy with device set to cuda:0

Returns a copy with device set to cuda:0

def cudaIndex[S : Sc](index: Short): STenOptions

Returns a copy with device set to cuda with index

Returns a copy with device set to cuda with index

def deviceIndex: Int
def isByte: Boolean
def isCPU: Boolean
def isCuda: Boolean
def isDouble: Boolean
def isFloat: Boolean
def isLong: Boolean
def isSparse: Boolean
def scalarTypeByte: Byte

Returns the byte representation of dtype

Returns the byte representation of dtype

0 - byte, 1 - char 2 - short 3 - int 4 - long, 5 - half 6 - float, 7 - double, 8 - complex half 9 - complex float 10 - complex double 11 - bool 15 - bfloat16 12,13,14,16 - quantized ints

Returns a copy with dtype set to a value compatible with Scala's Byte

Returns a copy with dtype set to a value compatible with Scala's Byte

Returns a copy with dtype set to double

Returns a copy with dtype set to double

Returns a copy with dtype set to float

Returns a copy with dtype set to float

Returns a copy with dtype set to long

Returns a copy with dtype set to long

Inherited methods

def productElementNames: Iterator[String]
Inherited from:
Product
def productIterator: Iterator[Any]
Inherited from:
Product