cudf.from_dlpack

cudf.from_dlpack(pycapsule_obj)

Converts from a DLPack tensor to a cuDF object.

DLPack is an open-source memory tensor structure: dmlc/dlpack.

This function takes a PyCapsule object which contains a pointer to a DLPack tensor as input, and returns a cuDF object. This function deep copies the data in the DLPack tensor into a cuDF object.

Parameters

pycapsule_objPyCapsule

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

Returns

A cuDF DataFrame or Series depending on if the input DLPack tensor is 1D

or 2D.

Notes

cuDF from_dlpack() assumes column-major (Fortran order) input. If the input tensor is row-major, transpose it before passing it to this function.