Skip to main content

Ctrl+K

Installation Guide
API Docs
- Metric
- Neural Network
  - Data structure
  - Layers
    
    BaseLayer
    
    Activations
    
    BatchNorm
    
    Conv2D
    
    ConvTranspose2D
    
    Embedding
    
    Layer Block
    
    LayerNorm
    
    Linear
    
    LSTM
    
    SLinear
    
    SLSTM
    
    Pooling
    
    ResNet Block
  - Sequential
  - Output Updater
  - Ddp
- Tagi_utils
- Cuda
Tutorials
Github Development Guide
Distributed Data Parallel
Theory
- Activation functions
- Delta updates
About py/cuTAGI

Layers

Layers#

BaseLayer
Activations
BatchNorm
Conv2D
ConvTranspose2D
Embedding
Layer Block
LayerNorm
Linear
LSTM
SLinear
SLSTM
Pooling
ResNet Block

previous

pytagi.nn.data_struct

next

pytagi.nn.base_layer

By Luong-Ha Nguyen, James-A. Goulet, Van Dai Vuong, Miquel Florensa-Montilla

© Copyright 2025, Luong-Ha Nguyen, James-A. Goulet, Van Dai Vuong, Miquel Florensa-Montilla.