Is there some research work or any function in pytorch which can be used to take the importance of a filter in CNN?