Skip to main content

Ctrl+K

TorchBrain

User Guide
API Reference

GitHub
Discord
PyPI

User Guide
API Reference

GitHub
Discord
PyPI

Section Navigation

torch_brain.dataset
torch_brain.data
- chain
- collate
- pad
- pad8
- pad2d
- pad2d8
- track_batch
- track_mask
- track_mask8
- track_mask2d
- track_mask2d8
- Dataset
torch_brain.data.sampler
torch_brain.transforms
torch_brain.nn
torch_brain.models
torch_brain.registry
torch_brain.utils

API Reference
torch_brain.transforms
RandomOutputSampler

RandomOutputSampler#

class torch_brain.transforms.RandomOutputSampler(num_output_tokens)[source]#: Bases: object

previous

RandomTimeScaling

next

RandomCrop

On this page

RandomOutputSampler

© Copyright 2026, neuro-galaxy Team.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.17.1.