OpaqueTensor#

class nvmath.device.OpaqueTensor(*args)[source]#

Abstraction over the cuBLASDx tensor type (an alias of the CuTe tensor type). The CuTe tensor layout is powerful and supports layouts not provided by NumPy, so this a bridge to add this functionality to Python.

Note

Do not create directly, use nvmath.device.make_tensor().

Refer to the cuBLASDx documentation for more details on how to use this class: https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#tensors

Methods

__init__(*args)[source]#

Attributes

buffer: ndarray#

layout: Layout#

leading_dimension: int | None#