wehub-resource-sync
11 lines
327 B
Python
11 lines
327 B
Python
"""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
|