I want to implement pruning by setting some weights to 0.0 (specific weights in each layer, selected in a way that is not important to discuss here) to (1) ignore them during in