cutensor-cu12 2.0.2

主要特性

• 广泛支持混合精度

  • FP64输入，FP32计算。

  • FP32输入，使用FP16、BF16或TF32进行计算。

  • 复数-实数运算。

  • 支持共轭（不进行转置）操作。

• 支持高达64维度的张量。

• 任意数据布局。

• 简单可序列化的数据结构。

• 主要计算例程

  • 直接（即，无转置）张量收缩。

    • 支持专用内核的即时编译。

  • 张量缩减（包括部分缩减）。

  • 元素级张量操作

    • 支持各种激活函数。

    • 支持输出张量的填充。

    • 任意张量排列。

    • 不同数据类型之间的转换。

文档

有关cuTENSOR文档，请参阅https://docs.nvda.net.cn/cuda/cutensor/index.html。

安装

cuTENSOR wheel可以按照以下方式安装

pip install cutensor-cuXX

其中XX是CUDA主要版本（目前支持CUDA 11 & 12）。包cutensor（不带-cuXX后缀）已弃用。如果您已安装cutensor，请先将其移除，然后再安装cutensor-cuXX。