|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
Additional optional constants.
|
|
"""
|
|
|
|
from qiskit.utils import LazyImportTester
|
|
|
|
|
|
HAS_TORCH = LazyImportTester(
|
|
{
|
|
"torch": ("cat", "einsum", "is_tensor", "nn", "optim", "sparse_coo_tensor", "Tensor"),
|
|
"torch.autograd": ("Function",),
|
|
"torch.autograd.variable": ("Variable",),
|
|
"torch.nn": (
|
|
"L1Loss",
|
|
"Linear",
|
|
"Module",
|
|
"MSELoss",
|
|
"Parameter",
|
|
),
|
|
"torch.nn.functional": (),
|
|
"torch.optim": (
|
|
"Adam",
|
|
"SGD",
|
|
),
|
|
"torch.utils.data": ("Dataset",),
|
|
},
|
|
name="PyTorch",
|
|
install="pip install 'qiskit-machine-learning[torch]'",
|
|
)
|
|
|
|
HAS_SPARSE = LazyImportTester(
|
|
{
|
|
"sparse": ("SparseArray", "COO", "DOK"),
|
|
},
|
|
name="sparse",
|
|
install="pip install 'qiskit-machine-learning[sparse]'",
|
|
)
|
|
|