docstr: Add pruning on the fly and reparametrization of a tensor.
  
  Adds the forward pre-hook that enables pruning on the fly and
  the reparametrization of a tensor in terms of the original tensor
  and the pruning mask.
  
  Args:
      module (nn.Module): module containing the tensor to prune
      name (str): parameter name within ``module`` on which pruning
          will act.
docstr_tok: ["Add","pruning","on","the","fly","and","reparametrization","of","a","tensor",".","Adds","the","forward","pre-hook","that","enables","pruning","on","the","fly","and","the","reparametrization","of","a","tensor","in","terms","of","the","original","tensor","and","the","pruning","mask",".","Args",":","module","(","nn.Module",")",":","module","containing","the","tensor","to","prune","name","(","str",")",":","parameter","name","within","``","module","``","on","which","pruning","will","act","."]
code: def apply(cls, module, name, mask):
      r"""Add pruning on the fly and reparametrization of a tensor.
  
      Adds the forward pre-hook that enables pruning on the fly and
      the reparametrization of a tensor in terms of the original tensor
      and the pruning mask.
  
      Args:
          module (nn.Module): module containing the tensor to prune
          name (str): parameter name within ``module`` on which pruning
              will act.
      """
      return super().apply(module, name, mask=mask)
  
imports: import numbers
  from abc import ABC, abstractmethod
  from collections.abc import Iterable
  from typing import Tuple
  import torch
label_desc: Use image node_id 3 for calling the CustomFromMask obj's underlying member method code with example usage: obj.apply(cls, module, name, mask) and returns: super