cutensor 1.6.0.3

主要特性

• 广泛的混合精度支持

  • FP64输入与FP32计算。

  • FP32输入与FP16、BF16或TF32计算。

  • 复数与实数运算。

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

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

• 任意数据布局。

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

• 主要计算例程

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

  • 张量 reduction（包括部分 reduction）。

  • 逐元素张量操作

    • 支持各种激活函数。

    • 任意张量排列。

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

文档

请参阅https://docs.nvda.net.cn/cuda/cutensor/index.html以获取 cuTENSOR 文档。