Method GetTensorDesc

dali/c_api_2/data_objects.h:688–698

  }

  daliTensorDesc_t GetTensorDesc(int sample) const override {
    ValidateSampleIdx(sample);
    daliTensorDesc_t desc{};
    auto &shape = tl_->shape();
    desc.ndim = shape.sample_dim();
    desc.data = tl_->raw_mutable_tensor(sample);
    desc.dtype = tl_->type();
    desc.layout = GetLayout();
    desc.shape = shape.tensor_shape_span(sample).data();
    return desc;
  }

  const TensorListShape<> &GetShape() const & override {
    return tl_->shape();