code: def __init__(self, in_features, out_features, bias=True):
      super(FusedDense, self).__init__()
      self.in_features = in_features
      self.out_features = out_features
      self.weight = nn.Parameter(torch.empty(out_features, in_features))
      if bias:
          self.bias = nn.Parameter(torch.empty(out_features))
      else:
          # assert False, "no-bias option not added yet"
          self.register_parameter("bias", None)
  
imports: import torch
  from torch import nn
  import fused_dense_cuda
  from apex._autocast_utils import _cast_if_autocast_enabled
label_desc: Use image node_id 1 to create a new FusedDense object from inherited base classes: nn with example: obj = FusedDense(in_features, out_features, bias)