cuTensorNet：用于张量网络计算的高性能库

欢迎来到 cuTensorNet 库文档！

NVIDIA cuTensorNet 是用于张量网络计算的高性能库，是 NVIDIA cuQuantum SDK 的组件。cuTensorNet 的功能在 概述 中描述，安装和使用指南在 快速入门 中提供。

主要功能

  • 基于 NVIDIA 的高性能张量代数库：cuTENSOR

  • 提供 API 用于

    • 创建张量或张量网络对象

    • 为任何给定的张量网络找到成本最优的张量网络收缩路径

    • 为张量网络收缩找到低开销的切片，以满足指定的内存约束

    • 调整张量网络收缩路径查找器配置以获得更好的性能

    • 执行张量网络收缩计划生成、自动调优及其后续执行

    • 逐步构建张量网络状态（例如，量子线路状态），然后计算其属性，包括幅度的任意切片、期望值、边际分布（约化密度矩阵）以及执行直接采样

    • 执行反向传播以计算输出张量相对于用户指定的输入张量的梯度

    • 使用 QR 或 SVD 执行张量分解

    • 将量子门操作数应用于一对连接（收缩）的张量

    • 在收缩路径查找器和执行器中启用自动分布式并行化

    • 启用自定义内存管理

    • 日志记录

支持

  • 支持的 GPU 架构：Volta、Turing、Ampere、Ada、Hopper

  • 支持的操作系统：Linux

  • 支持的 CPU 架构：x86_64、ARM64

先决条件

目录