cuquantum.cutensornet.network_set_attribute¶

cuquantum.cutensornet.network_set_attribute(intptr_t handle, intptr_t network_desc, int attr, intptr_t buf, size_t size_in_bytes)[source]¶

Sets attributes of network_descriptor.

Parameters

handle (intptr_t) – Opaque handle holding cuTensorNet’s library context.
network_desc (intptr_t) – Opaque structure that is accessed.
attr (NetworkAttribute) – Specifies the attribute that is requested.
buf (intptr_t) – This buffer (of size size_in_bytes) determines the value to which attr will be set.
size_in_bytes (size_t) – Size of buf (in bytes).

Note

To compute the attribute size, use the itemsize of the corresponding data type, which can be queried using get_network_attribute_dtype().