Skip to main content
Ctrl+K

Vision Architectures (main)

  • API Reference
  • API Reference

Section Navigation

  • Blocks
    • Squeeze and Excitation
    • Heads3D
    • MBConv3D
    • Transformer
    • CNN
  • Image Readers
    • SafeTensorsReader
  • Layers
    • Embeddings
    • Attention
    • Latent Space
    • Upsample / Downsample
    • Codebook
  • Loss Functions
    • ClassBalancedCrossEntropyLoss
  • Metrics
    • Detection
  • Nets
    • Perceiver3D
    • UperNet3D
    • DETR3D
    • MaxViT3D
    • FPN3D
    • SymSwin3D
    • SwinV23D
    • ViT3D
    • UNetR3DDecoder
    • UperNet2D
    • FPN2D
    • CaiT3D
    • Swin3D
  • Schedulers
    • Noise
    • Sigmoid
    • Cyclic
    • Learning Rate Schedulers
  • Transforms
    • CropPad
    • Clipping
    • Spatial
    • Resize
  • Utils
    • Clamping / Clipping
    • Bounding Boxes
    • Pipeline Parallelism
    • Residual Connections
    • Normalizations
    • Rearrange
    • ActivationCheckpointing
    • Tensor Splitter and Merger
    • CustomBaseModel
    • Exponential Moving Average (EMA)
    • Timesteps
    • Activations
  • API Reference
  • Layers

Layers#

This section documents the layers components.

  • Embeddings
  • Attention
  • Latent Space
  • Upsample / Downsample
  • Codebook

previous

SafeTensorsReader

next

Embeddings

© Copyright 2023 onwards, Arjun Agarwal.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.16.1.