cuquantum.cutensornet.get_output_tensor_descriptor

cuquantum.cutensornet.get_output_tensor_descriptor(intptr_t handle, intptr_t desc_net) → intptr_t

创建一个 cutensornetTensorDescriptor_t，表示网络的输出张量。

参数

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

• desc_net (intptr_t) – 指向 cutensornetNetworkDescriptor_t 的指针。

返回

一个不透明的 cutensornetTensorDescriptor_t 结构体。不能为空。返回时，新的 cutensornetTensorDescriptor_t 保存 desc_net 输出张量的元数据。

返回类型

intptr_t

另请参阅

cutensornetGetOutputTensorDescriptor()