Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
torch_brain documentation
Logo
torch_brain documentation

Get Started

  • Installation

Concepts

  • Sampling

Package Reference

  • torch_brain.data
    • Dataset
    • Collate
    • Samplers
  • torch_brain.nn
    • Embeddings Layers
    • Transformer modules
    • Losses
    • Readout Layers
  • torch_brain.models
  • torch_brain.registry
  • torch_brain.optim
  • torch_brain.transforms
  • torch_brain.utils
Back to top

torch_brain.dataΒΆ

  • Dataset
    • Dataset
    • DatasetIndex
  • Collate
    • collate()
    • pad()
    • track_mask()
    • pad8()
    • track_mask8()
    • chain()
    • track_batch()
  • Samplers
    • SequentialFixedWindowSampler
    • RandomFixedWindowSampler
    • TrialSampler
    • DistributedStitchingFixedWindowSampler
Next
Dataset
Previous
Sampling
Copyright © 2025, neuro-galaxy Team
Made with Sphinx and @pradyunsg's Furo