copy

nvmath.device.copy(

src: OpaqueTensor,

dst: OpaqueTensor,

alignment=None,

)

Copies data from the source tensor to the destination tensor.

Parameters:

• src (OpaqueTensor) – The source tensor to copy from.

• dst (OpaqueTensor) – The destination tensor to copy to.

Refer to the cuBLASDx documentation for more details on how to use this function: https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#cooperative-global-shared-copying