nvmath.device.TransposeMode

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

A namedtuple class that encapsulates the transpose mode for input matrices A and B in matrix multiplication.

a

The operation that needs to be performed with input matrix A, currently supports 'non_transposed', 'transposed' and 'conj_transposed'.

b

The operation that needs to be performed with input matrix B, currently supports 'non_transposed', 'transposed' and 'conj_transposed'.