I have tried to implement MaskedLinear.
↓↓↓

  
      
      GitHub
  
  
    

uchida-takumi/MaskedLinear 214

This is an extended torch.nn.Linear module which mask connection. - uchida-takumi/MaskedLinear


  
  
    
    
  
  


I am glad, if it help someone.