TorchConnector#
- class TorchConnector(*args, **kwargs)[source]#
Bases :
torch.nn.Module
Connects a Qiskit (Quantum) Neural Network to PyTorch.
- Paramètres:
neural_network (NeuralNetwork) – The neural network to be connected to PyTorch. Remember that
input_gradients
must be set toTrue
in the neural network initialization before passing it to theTorchConnector
for the gradient computations to work properly during training.initial_weights (np.ndarray | Tensor | None) – The initial weights to start training the network. If this is None, the initial weights are chosen uniformly at random from [-1, 1].
sparse (bool | None) – Whether this connector should return sparse output or not. If sparse is set to None, then the setting from the given neural network is used. Note that sparse output is only returned if the underlying neural network also returns sparse output, otherwise an error will be raised.
- Lève:
QiskitMachineLearningError – If the connector is configured as sparse and the underlying network is not sparse.
Attributes
- neural_network#
Returns the underlying neural network.
- sparse#
Returns whether this connector returns sparse output or not.
- weight#
Returns the weights of the underlying network.
Methods