nvmath.device.LeadingDimension

class nvmath.device.LeadingDimension(a, b, c)[source]

A namedtuple class that encapsulates the three leading dimensions in matrix multiplication \(C = \alpha Op(A) Op(B) + \beta C\).

a

The leading dimension of two-dimensional array used to store the matrix A.

Type:

int

b

The leading dimension of two-dimensional array used to store the matrix B.

Type:

int

c

The leading dimension of two-dimensional array used to store the matrix C.

Type:

int