Weight Layer from paper

Hello! I want to implements Deep Feature Selection method from this paper: https://pdfs.semanticscholar.org/ac2a/c075773cc936a206c3ebb2339376d586bcbd.pdf

But I can’t understand how I can make Weight Layer after input layer? Is it just nn.Linear or smth else? Thx!