If i follow this solution I can deactivate only all the weights at the same time but not a specific element of the weight matrix. At the end of the day i want to have 2 layers where the connections are sparse, so y = Wx where W the weight matrix with some elements zero during the whole training process