Skip to content

Instantly share code, notes, and snippets.

@rdisipio
Created December 17, 2020 22:00
Show Gist options
  • Save rdisipio/885fcef206710fcfd4213446033270fb to your computer and use it in GitHub Desktop.
Save rdisipio/885fcef206710fcfd4213446033270fb to your computer and use it in GitHub Desktop.
qlstm_qlayer
n_qubits = 4
dev = qml.device("default.qubit", wires=n_qubits)
def _circuit(inputs, weights):
qml.templates.AngleEmbedding(inputs, wires=range(n_qubits))
qml.templates.BasicEntanglerLayers(weights, wires=range(n_qubits))
return [qml.expval(qml.PauliZ(wires=i)) for i in range(n_qubits)]
qlayer = qml.QNode(_circuit, dev, interface="torch")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment