Return type:
      module (nn.Module)
-[ Examples ]-



from torch.nn.utils import prune
m = prune.custom_from_mask(
   ...     nn.Linear(5, 3), name='bias', mask=torch.tensor([0, 1, 0])
   ... )
print(m.bias_mask)
   tensor([0., 1., 0.])


