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, ora 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.
See also