cuquantum.cutensornet.network_get_attribute

cuquantum.cutensornet.network_get_attribute(intptr_t handle, intptr_t network_desc, int attr, intptr_t buf, size_t size_in_bytes)[源代码]

获取 network_descriptor 的属性。

参数

  • handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。

  • network_desc (intptr_t) – 要访问的不透明结构。

  • attr (NetworkAttribute) – 指定请求的属性。

  • buf (intptr_t) – 返回时，此缓冲区（大小为 size_in_bytes）保存与 network_desc 中的 attr 对应的值。

  • size_in_bytes (size_t) – buf 的大小（以字节为单位）。

注意

要计算属性大小，请使用相应数据类型的 itemsize，可以使用 get_network_attribute_dtype() 查询。

另请参阅

cutensornetNetworkGetAttribute()