Layout

class nvmath.device.Layout

Layout for the nvmath.device.OpaqueTensor.

Note

Do not create directly, use appropriate method from nvmath.device.Matmul(). Refer to https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#imported-tensor-utilities for guidance on which method to use.

Methods

__init__()

Attributes

cosize

Returns a distance from last element of a tensor to its first element. It describes how many elements does the argument layout span.

Refer to the cuBLASDx documentation for more details on how to use this attribute: https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#imported-tensor-utilities

size

Number of valid elements in a tensor. This is simply a product of all shape dimensions.

Refer to the cuBLASDx documentation for more details on how to use this attribute: https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#imported-tensor-utilities