def _prune_weights(self, indices: Tensor):
        """Given a list of indices, prune the weights at those indices.

        Pruning simply marks the weight as pruned in the `task_index` and
        makes the weight invisible in the `visible_mask`.

        :param indices: A 1D tensor of indices to prune
        """
        self.task_index.flatten()[indices] = self.PRUNED_CODE.item()
        self.visible_mask.flatten()[indices] = False
Given a list of indices, prune the weights at those indices.

        Pruning simply marks the weight as pruned in the `task_index` and
        makes the weight invisible in the `visible_mask`.

        :param indices: A 1D tensor of indices to prune
        
avalanche/models/packnet.py