Skip to main content
Back to top
Ctrl
+
K
TorchBrain
User Guide
API Reference
Search
Ctrl
+
K
GitHub
Discord
PyPI
Search
Ctrl
+
K
User Guide
API Reference
GitHub
Discord
PyPI
Collapse Sidebar
Expand Sidebar
Section Navigation
torch_brain.dataset
Dataset
DatasetIndex
NestedDataset
SpikingDatasetMixin
CalciumImagingDatasetMixin
MultiChannelDatasetMixin
torch_brain.data
chain
collate
pad
pad8
pad2d
pad2d8
track_batch
track_mask
track_mask8
track_mask2d
track_mask2d8
Dataset
torch_brain.data.sampler
RandomFixedWindowSampler
SequentialFixedWindowSampler
TrialSampler
DistributedEvaluationSamplerWrapper
DistributedStitchingFixedWindowSampler
torch_brain.transforms
Compose
RandomChoice
ConditionalChoice
UnitDropout
TriangleDistribution
RandomTimeScaling
RandomOutputSampler
RandomCrop
BinSpikes
UnitFilter
UnitFilterById
torch_brain.nn
Embedding
InfiniteVocabEmbedding
RotaryTimeEmbedding
SinusoidalTimeEmbedding
RotaryCrossAttention
RotarySelfAttention
MultitaskReadout
prepare_for_multitask_readout
Loss
MSELoss
CrossEntropyLoss
MallowDistanceLoss
torch_brain.models
POYO
POYOPlus
CalciumPOYOPlus
torch_brain.registry
DataType
ModalitySpec
register_modality
get_modality_by_id
MODALITY_REGISTRY
torch_brain.utils
stitch
seed_everything
create_linspace_latent_tokens
create_start_end_unit_tokens
resolve_weights_based_on_interval_membership
isin_interval
prepare_for_readout
np_string_prefix
bin_spikes
API Reference
torch_brain.registry
MODALITY_REGISTRY
MODALITY_REGISTRY
#
dict() -> new empty dictionary
