Method ReshapeTensor

python/mod_cvcuda/nvcv/Tensor.cpp:185–200

}

std::shared_ptr<Tensor> Tensor::ReshapeTensor(Tensor &tensor, Shape shape, std::optional<nvcv::TensorLayout> layout)
{
    Tensor::Key key;
    Cache::Instance().removeAllNotInUseMatching(key);

    nvcv::Tensor tensor_impl      = tensor.impl();
    auto         new_tensor_shape = CreateNVCVTensorShape(shape, layout ? *layout : tensor_impl.layout());
    nvcv::Tensor new_tensor_impl  = tensor_impl.reshape(std::move(new_tensor_shape));
    auto         new_tensor       = std::shared_ptr<Tensor>(new Tensor(std::move(new_tensor_impl)));

    // Need to add wrappers to cache so that they don't get destroyed by
    // the cuda stream when they're last used, and python script isn't
    // holding a reference to them. If we don't do it, things might break.
    Cache::Instance().add(*new_tensor);
    return new_tensor;
}

std::shared_ptr<Tensor> Tensor::Reshape(Shape shape, std::optional<nvcv::TensorLayout> layout)
{