How to compute the normal vectors¿

Hi! Your work is amazing. I am trying to extend it to be able to extract normal vectors for a given trained model.
Do you have any suggested approach on how to do it?
Thanks in advance!