"""Feature storage classes for DataLoading"""
from .. import backend as F

from .base import *
from .numpy import *
# Defines the name TensorStorage
if F.get_preferred_backend() == 'pytorch':
    from .pytorch_tensor import PyTorchTensorStorage as TensorStorage
else:
    from .tensor import BaseTensorStorage as TensorStorage
