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.nn
prepare_for_multitask_readout

prepare_for_multitask_readout#

torch_brain.nn.prepare_for_multitask_readout(data, readout_registry)[source]#

previous

MultitaskReadout

next

Loss

On this page

prepare_for_multitask_readout()

© Copyright 2026, neuro-galaxy Team.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.17.1.