@qml.qnode(dev)
def circuit():
    qml.QubitStateVector(rnd_state, wires=range(n_wires))
    qml.QubitUnitary(mat, wires=list(range(n_wires))).inv()
    return qml.probs(wires=range(n_wires))
@qml.qnode(dev)
def circuit():
    qml.QubitStateVector(rnd_state, wires=range(n_wires))
    qml.QubitUnitary(mat, wires=list(range(n_wires))).inv()
    return qml.probs(wires=range(n_wires))