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().