Method WrapExternalBufferVector

python/mod_cvcuda/nvcv/TensorBatch.cpp:62–80

}

std::shared_ptr<TensorBatch> TensorBatch::WrapExternalBufferVector(std::vector<py::object>           buffers,
                                                                   std::optional<nvcv::TensorLayout> layout)
{
    TensorList list;
    list.reserve(buffers.size());
    for (auto &obj : buffers)
    {
        std::shared_ptr<ExternalBuffer> buffer = cast_py_object_as<ExternalBuffer>(obj);
        if (!buffer)
        {
            throw std::runtime_error("Input buffer doesn't provide cuda_array_interface or DLPack interfaces.");
        }
        auto tensor = Tensor::Wrap(*buffer, layout);
        list.push_back(tensor);
    }
    auto batch = Create(buffers.size());
    batch->pushBackMany(list);
    return batch;
}

TensorBatch::TensorBatch(int capacity)
    : m_key(capacity)