Function GetDLTensorResource

dali/operators/python_function/dltensor_function.cc:215–226

using DLTensorNumpyResource = DLTensorResource<PyArrayPayload>;

auto GetDLTensorResource(const py::array &array) {
  auto rsrc = DLTensorNumpyResource::Create(array);
  auto &tensor = rsrc->dlm_tensor.dl_tensor;
  auto buffer = rsrc->payload.array.request();
  tensor.data = buffer.ptr;
  tensor.shape = rsrc->payload.shape.data();
  tensor.ndim = rsrc->payload.shape.size();
  tensor.strides = rsrc->payload.strides.empty() ? nullptr : rsrc->payload.strides.data();
  tensor.device = ToDLDevice(false, false, 0);
  tensor.dtype = ToDLType(TypeFromFormatStr(buffer.format).id());
  return rsrc;
}


PYBIND11_MODULE(python_function_plugin, m, py::mod_gil_not_used()) {