Function TensorBatchHQResize

python/mod_cvcuda/operators/OpHQResize.cpp:519–545

}

TensorBatch TensorBatchHQResize(TensorBatch &src, const std::vector<Shape> &outShape, std::optional<bool> antialias,
                                const std::optional<Rois> &roi, std::optional<NVCVInterpolationType> interpolation,
                                std::optional<NVCVInterpolationType> minInterpolation,
                                std::optional<NVCVInterpolationType> magInterpolation, std::optional<Stream> pstream)
{
    TensorBatch out = TensorBatch::Create(src.numTensors());

    int32_t numOutSizes = outShape.size();
    if (numOutSizes != src.numTensors() && numOutSizes != 1)
    {
        throw std::runtime_error(
            "The list of output shapes `out_size` must either contain a single shape to be used for all output tensors "
            "or its length must match the number of input tensors.");
    }

    for (int i = 0; i < src.numTensors(); ++i)
    {
        auto        sampleShape  = outShape[numOutSizes == 1 ? 0 : i];
        const auto &inSample     = src[i];
        auto        resizedShape = ResizedTensorShape(inSample.layout(), inSample.shape(), sampleShape);
        Tensor      dst          = Tensor::Create(resizedShape, src.dtype(), src.layout());
        out.pushBack(dst);
    }

    return TensorBatchHQResizeInto(out, src, antialias, roi, interpolation, minInterpolation, magInterpolation,
                                   pstream);
}

} // namespace