input_mask 来自  It's mask input preprocessed and outputted by plugin embLayerNormPlugin    
https://github.com/NVIDIA/TensorRT/tree/main/plugin/embLayerNormPlugin

https://github.com/lix19937/dnn-cookbook/blob/main/ViT-pytorch/generate_mha.py

https://zhuanlan.zhihu.com/p/517782166

https://github.com/lix19937/history/tree/main/zz/python/trt-plugin/bertqkvtocontextplugin     

https://github.com/NVIDIA/TensorRT/issues/3831#issuecomment-2089430724    
https://github.com/NVIDIA/TensorRT/issues/2601   
https://github.com/NVIDIA/TensorRT/issues/2655   
