How to make youcompleteme work for libtorch

I’m new to libtorch, and I want to use it on ubuntu16.04 platform with vim editor. I have added /usr/local/libtorch/include/torch/csrc/api/include/ to the flags in .ycm_extra_conf.py, but the vim editor can only complete part of the content like types and cuda related. Important things like torch::nn can not be completed. I want to get the correct configure for youcompleteme plugin in vim editor. Wish your help!

The complete like follows: