docs
0.0-2dfa388-SNAPSHOT
docs
API
torchvision
datasets
FashionMNIST
MNIST
MNISTBase
models
resnet
BasicBlock
BasicBlock
BlockBuilder
Bottleneck
Bottleneck
ResNet
ResNet101
ResNet152
ResNet18
ResNet34
ResNet50
ResNetFactory
ResNetVariant
Weights
transforms
Presets
ImageClassification
functional
torch
cuda
data
DataLoader
Example
ExampleDataLoader
TensorDataset
TensorDataset
TensorSeq
TupleDataLoader
nn
functional
loss
CrossEntropyLoss
modules
activation
LogSoftmax
ReLU
Softmax
Tanh
batchnorm
BatchNorm1d
BatchNorm2d
container
ModuleList
Sequential
conv
Conv2d
Conv2d
PaddingMode
flatten
Flatten
linear
Identity
Linear
normalization
GroupNorm
LayerNorm
pooling
AdaptiveAvgPool2d
MaxPool2d
regularization
Dropout
sparse
Embedding
HasParams
HasWeight
Module
TensorModule
TensorModuleBase
init
Mode
NonLinearity
utils
optim
lr_scheduler
LRScheduler
StepLR
Adam
AdamW
Optimizer
SGD
special
BFloat16
BFloat16Tensor
Bits16
Bits16Tensor
Bits1x8
Bits1x8Tensor
Bits2x4
Bits2x4Tensor
Bits4x2
Bits4x2Tensor
Bits8
Bits8Tensor
Bool
BoolTensor
Complex128
Complex128Tensor
Complex32
Complex32Tensor
Complex64
Complex64Tensor
DType
DType
bfloat16
bits16
bits1x8
bits2x4
bits4x2
bits8
bool
complex128
complex32
complex64
float16
float32
float64
float8_e4m3fn
float8_e5m2
int16
int32
int64
int8
numoptions
qint32
qint8
quint2x4
quint4x2
quint8
uint8
undefined
Default
Default
Derive
Device
Device
DeviceType
DeviceType
Float16
Float16Tensor
Float32
Float32Tensor
Float64
Float64Tensor
Float8_e4m3fn
Float8_e4m3fnTensor
Float8_e5m2
Float8_e5m2Tensor
Generator
Int16
Int16Tensor
Int32
Int32Tensor
Int64
Int64Tensor
Int8
Int8Tensor
Layout
Layout
LowPriorityDefaults
MemoryFormat
NumOptions
NumOptionsTensor
QInt32
QInt32Tensor
QInt8
QInt8Tensor
QUInt2x4
QUInt2x4Tensor
QUInt4x2
QUInt4x2Tensor
QUInt8
QUInt8Tensor
Tensor
Tensor
TensorTuple
UInt8
UInt8Tensor
Undefined
UndefinedTensor
hub
indexing
Ellipsis
Ellipsis
Slice
Slice
docs
/
torch
/
Device
Device
torch.Device
See the
Device
companion object
case
class
Device
(
device
:
DeviceType
,
index
:
Byte
)
Attributes
Companion
object
Source
Device.scala
Graph
Reset zoom
Hide graph
Show graph
Supertypes
trait
Serializable
trait
Product
trait
Equals
class
Object
trait
Matchable
class
Any
Show all
Members list
Clear all
Value members
Inherited methods
def
productElementNames
:
Iterator
[
String
]
Attributes
Inherited from:
Product
def
productIterator
:
Iterator
[
Any
]
Attributes
Inherited from:
Product
In this article
Attributes
Members list
Value members
Inherited methods