nvmath.device.TransposeMode¶
- class nvmath.device.TransposeMode(a, b)[source]¶
A namedtuple class that encapsulates the transpose mode for input matrices
A
andB
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'
.