Created by: wanghaoshuang

Support custom functions when computing pruned substructure of parameter based on https://github.com/PaddlePaddle/PaddleSlim/pull/125