hipdf.DataFrame.to_dlpack

19 min read time

Applies to Linux

DataFrame.to_dlpack()

Converts a cuDF object to a DLPack tensor. DLPack is an open-source memory tensor structure: dmlc/dlpack.

Returns

PyCapsule

A DLPack tensor pointer which is encapsulated in a PyCapsule object.

Notes

The result is in column-major (Fortran order) format. If the output tensor needs to be row major, transpose the output of this function.