clear

nvmath.device.clear(arr: OpaqueTensor)

Clears the contents of the given tensor by setting all elements to zero.

Parameters:

arr (OpaqueTensor) – The tensor to be cleared.

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