network_set_input_tensor_memory#

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

Provide memory buffer and strides corresponding to an input tensor for the network to be used for data reading.

Parameters:

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

  • network_desc (intptr_t) – The network descriptor.

  • tensor_id (int64_t) – The tensor_id as returned by network_append_tensor().

  • 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

cutensornetNetworkSetInputTensorMemory()