Function DLTensorToCapsule

dali/util/pybind.h:214–217

Steal a DLPack Tensor from the passed pointer and wrap it into Python capsule.

// Steal a DLPack Tensor from the passed pointer and wrap it into Python capsule.
static py::capsule DLTensorToCapsule(DLMTensorPtr dl_tensor) {
  auto caps = py::capsule(dl_tensor.release(), DLTENSOR_NAME, &DLTensorCapsuleDestructor);
  return caps;
}

template <typename Backend>
py::capsule TensorToDLPackView(SampleView<Backend> tensor, bool pinned, int device_id) {