Update only a slice of the weight matrix at a time