class_docstr: Masked Linear Layer inheriting from `~torch.nn.Linear` class and applying a mask to consider
  only a selection of weight.
class_docstr_tok: ["Masked","Linear","Layer","inheriting","from","`","~torch.nn.Linear","`","class","and","applying","a","mask","to","consider","only","a","selection","of","weight","."]
code: def forward(self, x):
      return F.linear(x, self.mask * self.weight, self.bias)
  
file_path: benchmark_VAE/src/pythae/models/normalizing_flows/layers.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from pythae.models.base.base_utils import ModelOutput
label_desc: Use image node_id 2 for calling the MaskedLinear obj's underlying member method code with example usage: obj.forward(x) and returns: F