TransposeMode

class nvmath.device.TransposeMode(a, b)

一个 namedtuple 类，封装了矩阵乘法中输入矩阵 A 和 B 的转置模式。

a

需要对输入矩阵 A 执行的操作，目前支持 'non_transposed'、'transposed' 和 'conj_transposed'。

b

需要对输入矩阵 B 执行的操作，目前支持 'non_transposed'、'transposed' 和 'conj_transposed'。