cuquantum.cutensornet.marginal_compute¶

cuquantum.cutensornet.marginal_compute(intptr_t handle, intptr_t tensor_network_marginal, projected_mode_values, intptr_t work_desc, intptr_t marginal_tensor, intptr_t cuda_stream)[source]¶

Computes the requested tensor network state marginal tensor.

Parameters

handle (intptr_t) – cuTensorNet library handle.
tensor_network_marginal (intptr_t) – Tensor network state marginal representation.
projected_mode_values (object) –
Pointer to the values of the projected modes. Each integer value corresponds to a basis state of the given (projected) state mode. It can be:
- an int as the pointer address to the array, or
- a Python sequence of int64_t.
work_desc (intptr_t) – Workspace descriptor (the required scratch/cache memory buffers must be set by the user).
marginal_tensor (intptr_t) – Pointer to the GPU storage of the marginal tensor which will be computed in this call.
cuda_stream (intptr_t) – CUDA stream.