mmclassification/mmpretrain/structures/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .data_sample import DataSample
from .multi_task_data_sample import MultiTaskDataSample
from .utils import (batch_label_to_onehot, cat_batch_labels, label_to_onehot,
                    tensor_split)

__all__ = [
    'DataSample', 'batch_label_to_onehot', 'cat_batch_labels', 'tensor_split',
    'MultiTaskDataSample', 'label_to_onehot'
]