OpaqueTensor

class nvmath.device.OpaqueTensor(*args)

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: api/other_tensors.html#tensors

Attributes

buffer: ndarray

layout: Layout

leading_dimension: int | None

Methods