FateWithShellmiaoComment/Fate/python/federatedml/nn/backend/torch/__init__.py

try:
    from federatedml.nn.backend.torch import nn, init, operation, optim, serialization
except ImportError:
    nn, init, operation, optim, serialization = None, None, None, None, None

__all__ = ['nn', 'init', 'operation', 'optim', 'serialization']