def __init__(self, input_dim, output_dim, attention_norm_type, output_dropout_prob):
    super(SketchOutput, self).__init__()
    self.fc = nn.Linear(input_dim, output_dim)
    if (attention_norm_type not in NORM2FN):
        raise ValueError('The attention normalization is not in standard normalization types.')
    self.norm = NORM2FN[attention_norm_type](output_dim)
    self.dropout = nn.Dropout(output_dropout_prob)
def __init__(self, input_dim, output_dim, attention_norm_type, output_dropout_prob):
    super(SketchOutput, self).__init__()
    self.fc = nn.Linear(input_dim, output_dim)
    if (attention_norm_type not in NORM2FN):
        raise ValueError('The attention normalization is not in standard normalization types.')
    self.norm = NORM2FN[attention_norm_type](output_dim)
    self.dropout = nn.Dropout(output_dropout_prob)