LeadingDimension

class nvmath.device.LeadingDimension(a, b, c)

一个 namedtuple 类，封装了矩阵乘法 \(C = \alpha Op(A) Op(B) + \beta C\) 中的三个主导维度。

a

用于存储矩阵 A 的二维数组的主导维度。

类型:

int

b

用于存储矩阵 B 的二维数组的先行维度。

类型:

int

c

用于存储矩阵 C 的二维数组的先行维度。

类型:

int