Function TensorListGetItemImpl

dali/python/backend_impl.cc:1192–1209

template <typename Backend>
std::unique_ptr<Tensor<Backend> > TensorListGetItemImpl(TensorList<Backend> &t, Index id) {
  int num_tensors = static_cast<int>(t.num_samples());
  if (id < 0) {
    id = num_tensors + id;
  }
  if (id >= num_tensors || id < 0) {
      throw py::index_error("TensorListCPU index out of range");
  }
  auto ptr = std::make_unique<Tensor<Backend>>();
  // TODO(klecki): Rework this with proper sample-based tensor batch data structure
  auto &sample_shared_ptr = unsafe_sample_owner(t, id);
  auto &tshape = t.tensor_shape(id);
  size_t num_bytes = tshape.num_elements() * t.type_info().size();
  ptr->ShareData(sample_shared_ptr, num_bytes, t.is_pinned(), tshape, t.type(),
                 t.device_id(), t.order(), t.ready_event());
  ptr->SetMeta(t.GetMeta(id));
  return ptr;
}

template <typename Backend>
constexpr uint32_t TensorListDLPackRangeColor() {