Struct TQDevice 

pub struct TQDevice {
    pub n_wires: usize,
    pub device_name: String,
    pub bsz: usize,
    pub states: ArrayD<CType>,
    pub record_op: bool,
    pub op_history: Vec<OpHistoryEntry>,
}

Quantum device containing the quantum state vector

This struct mirrors TorchQuantum’s QuantumDevice class, providing:

• Multi-dimensional state tensor representation
• Batch support for parallel execution
• State reset and cloning operations

Fields

n_wires: usize

Number of qubits

device_name: String

Device name

bsz: usize

Batch size

states: ArrayD<CType>

Quantum state vector (batched, multi-dimensional)

record_op: bool

Whether to record operations

op_history: Vec<OpHistoryEntry>

Operation history

Implementations

impl TQDevice

pub fn new(n_wires: usize) -> Self

Create new quantum device

pub fn with_batch_size(n_wires: usize, bsz: usize) -> Self

Create quantum device with batch size

pub fn reset_states(&mut self, bsz: usize)

Reset to |0…0> state

pub fn reset_identity_states(&mut self)

Reset to identity matrix (useful for computing unitaries)

pub fn reset_all_eq_states(&mut self, bsz: usize)

Reset to equal superposition state

pub fn clone_states(&mut self, other: &TQDevice)

Clone states from another device

pub fn set_states(&mut self, states: ArrayD<CType>)

Set states directly

pub fn get_states_1d(&self) -> Array2<CType>

Get states as 1D vectors (shape: [bsz, 2^n_wires])

pub fn get_probs_1d(&self) -> Array2<f64>

Get probabilities (|amplitude|^2) as 1D vectors

pub fn record_operation(&mut self, entry: OpHistoryEntry)

Record an operation in history

pub fn reset_op_history(&mut self)

Clear operation history

pub fn apply_single_qubit_gate( &mut self, wire: usize, matrix: &Array2<CType>, ) -> Result<()>

Apply a single-qubit gate matrix to specified wire

pub fn apply_two_qubit_gate( &mut self, wire0: usize, wire1: usize, matrix: &Array2<CType>, ) -> Result<()>

Apply a two-qubit gate matrix to specified wires

pub fn apply_multi_qubit_gate( &mut self, wires: &[usize], matrix: &Array2<CType>, ) -> Result<()>

Apply a multi-qubit gate matrix to specified wires (n-qubit gate)

Trait Implementations

impl Clone for TQDevice

fn clone(&self) -> TQDevice

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TQDevice

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.