Operaciones básicas (adición, resta, multiplicación y división, multiplicación de matriz)

Transformación de la forma (remodelar/ver/exprimir/Unsqueezar)

Conversión de tipo de datos (Float32/int64, etc.)

Aceleración de GPU (.cuda ()/. To (dispositivo))

Propiedad requerida_grad

Backward () Backpropagation

Cálculo de gradiente de graduación

Capa básica

recipiente

Clase abstracta del conjunto de datos

Cargador de datos dataLoader

Conjuntos de datos predefinidos (TorchVision.Datasets)

Sgd

Adán

Rmsprop

Programador de tarifas de aprendizaje (LR_SCHEDULER)

MSE (error cuadrado medio)

CrossEntropy (entropía cruzada)

BCEWithLogits (clasificación binaria)

Función de pérdida personalizada

Modelo Save/Load (Torch.save ()/Torch.Load ())

Administración de dispositivos (TORCH.CUDA/IS_AVAILABLE ())

Configuración de semillas aleatorias (manual_seed)

DataParallel (tarjeta múltiple independiente)

DistributedDataparallel (múltiples máquinas y múltiples tarjetas)

Torchscript (serialización del modelo)

Exportación de ONNX