Index _ | A | B | C | D | E | F | G | I | L | M | N | P | R | S | T | U _ __getitem__() (torch_brain.dataset.Dataset method) (torch_brain.dataset.NestedDataset method) A apply_namespace() (torch_brain.dataset.Dataset method) B bin_spikes() (in module torch_brain.utils) BinSpikes (class in torch_brain.transforms) C CalciumImagingDatasetMixin (class in torch_brain.dataset) chain() (in module torch_brain.batching) collate() (in module torch_brain.batching) Compose (class in torch_brain.transforms) compute_average_firing_rates() (torch_brain.dataset.SpikingDatasetMixin method) ConditionalChoice (class in torch_brain.transforms) create_linspace_latent_tokens() (in module torch_brain.utils) create_start_end_unit_tokens() (in module torch_brain.utils) D Dataset (class in torch_brain.data) (class in torch_brain.dataset) DatasetIndex (class in torch_brain.dataset) datasets (torch_brain.dataset.NestedDataset property) detokenizer() (torch_brain.nn.InfiniteVocabEmbedding method) disable_data_leakage_check() (torch_brain.data.Dataset method) DistributedEvaluationSamplerWrapper (class in torch_brain.samplers) DistributedStitchingFixedWindowSampler (class in torch_brain.samplers) E Embedding (class in torch_brain.nn) extend_vocab() (torch_brain.nn.InfiniteVocabEmbedding method) extra_repr() (torch_brain.nn.InfiniteVocabEmbedding method) F forward() (torch_brain.models.POYO method) (torch_brain.nn.InfiniteVocabEmbedding method) (torch_brain.nn.RotaryCrossAttention method) (torch_brain.nn.RotarySelfAttention method) (torch_brain.nn.RotaryTimeEmbedding method) (torch_brain.nn.SinusoidalTimeEmbedding method) forward_varlen() (torch_brain.nn.RotaryCrossAttention method) (torch_brain.nn.RotarySelfAttention method) G get() (torch_brain.data.Dataset method) get_brainset_ids() (torch_brain.data.Dataset method) get_channel_ids() (torch_brain.dataset.MultiChannelDatasetMixin method) get_recording() (torch_brain.dataset.Dataset method) (torch_brain.dataset.NestedDataset method) get_recording_config_dict() (torch_brain.data.Dataset method) get_recording_data() (torch_brain.data.Dataset method) get_recording_hook() (torch_brain.dataset.Dataset method) get_roi_ids() (torch_brain.dataset.CalciumImagingDatasetMixin method) get_sampling_intervals() (torch_brain.data.Dataset method) (torch_brain.dataset.Dataset method) (torch_brain.dataset.NestedDataset method) get_session_ids() (torch_brain.data.Dataset method) get_subject_ids() (torch_brain.data.Dataset method) get_unit_ids() (torch_brain.data.Dataset method) (torch_brain.dataset.SpikingDatasetMixin method) I InfiniteVocabEmbedding (class in torch_brain.nn) init_vocabs() (torch_brain.models.POYO method) initialize_vocab() (torch_brain.nn.InfiniteVocabEmbedding method) invert() (torch_brain.nn.RotaryTimeEmbedding static method) is_lazy() (torch_brain.nn.InfiniteVocabEmbedding method) isin_interval() (in module torch_brain.utils) L load_pretrained() (torch_brain.models.POYO class method) M module torch_brain.batching torch_brain.data torch_brain.dataset torch_brain.models torch_brain.nn torch_brain.samplers torch_brain.transforms torch_brain.utils MultiChannelDatasetMixin (class in torch_brain.dataset) N NestedDataset (class in torch_brain.dataset) np_string_prefix() (in module torch_brain.utils) P pad() (in module torch_brain.batching) pad2d() (in module torch_brain.batching) pad2d8() (in module torch_brain.batching) pad8() (in module torch_brain.batching) POYO (class in torch_brain.models) R RandomChoice (class in torch_brain.transforms) RandomCrop (class in torch_brain.transforms) RandomFixedWindowSampler (class in torch_brain.samplers) RandomOutputSampler (class in torch_brain.transforms) RandomTimeScaling (class in torch_brain.transforms) rank_len() (torch_brain.samplers.DistributedEvaluationSamplerWrapper method) recording_ids (torch_brain.dataset.Dataset property) reset_parameters() (torch_brain.nn.Embedding method) (torch_brain.nn.InfiniteVocabEmbedding method) resolve_weights_based_on_interval_membership() (in module torch_brain.utils) RotaryCrossAttention (class in torch_brain.nn) RotarySelfAttention (class in torch_brain.nn) RotaryTimeEmbedding (class in torch_brain.nn) rotate() (torch_brain.nn.RotaryTimeEmbedding static method) S sequence_index (torch_brain.samplers.DistributedStitchingFixedWindowSampler attribute) SequentialFixedWindowSampler (class in torch_brain.samplers) set_epoch() (torch_brain.samplers.DistributedStitchingFixedWindowSampler method) set_params() (torch_brain.samplers.DistributedEvaluationSamplerWrapper method) SinusoidalTimeEmbedding (class in torch_brain.nn) SpikingDatasetMixin (class in torch_brain.dataset) stitch() (in module torch_brain.utils) subset_vocab() (torch_brain.nn.InfiniteVocabEmbedding method) T tokenize() (torch_brain.models.POYO method) tokenizer() (torch_brain.nn.InfiniteVocabEmbedding method) torch_brain.batching module torch_brain.data module torch_brain.dataset module torch_brain.models module torch_brain.nn module torch_brain.samplers module torch_brain.transforms module torch_brain.utils module track_batch() (in module torch_brain.batching) track_mask() (in module torch_brain.batching) track_mask2d() (in module torch_brain.batching) track_mask2d8() (in module torch_brain.batching) track_mask8() (in module torch_brain.batching) TrialSampler (class in torch_brain.samplers) TriangleDistribution (class in torch_brain.transforms) U UnitDropout (class in torch_brain.transforms) UnitFilter (class in torch_brain.transforms) UnitFilterById (class in torch_brain.transforms)
