network_set_output_tensor_memory#

cuquantum.bindings.cutensornet.network_set_output_tensor_memory(
intptr_t handle,
intptr_t network_desc,
intptr_t buffer,
strides,
)[source]#

Provide memory buffer and strides corresponding to the output tensor of the network to be used for data writing.

Parameters:

  • handle (intptr_t) – Opaque handle holding cuTensorNet’s library context.

  • network_desc (intptr_t) – The network descriptor.

  • buffer (intptr_t) – Pointer to memory buffer in device memory.

  • strides (object) –

    Array of size the number of modes of the corresponding tensor; strides[j] corresponding to the linearized offset – in physical memory – between two logically-neighboring elements w.r.t the j-th mode of the tensor. It can be:

    • an int as the pointer address to the array, or

    • a Python sequence of int64_t.

See also

cutensornetNetworkSetOutputTensorMemory()