How to customize the head intital weight with DGL

I am a biology researcher. In my project, for better interpretability, I want to customize multiple-heads between nodes. What I want: the number of heads is the number of actual interactions between nodes (this is easy to implement with DGL), and the head weight is the actual interaction force (a numerical value). I want to know if I can customize the initial weight of the head (I may not find it in the documentation).
Can somebody can help me? Thanks!! :heart:

Could you explain the details? What’s the head? and which part you’d like to customize?