def __init__(self, num_heads, hidden_dim, inter_dim, attention_norm_type, inter_activation, attention_dropout_prob, hidden_dropout_prob, output_dropout_prob):
    super(SketchLayer, self).__init__()
    self.attention = SketchMultiHeadAttention(num_heads, hidden_dim, attention_norm_type, attention_dropout_prob, hidden_dropout_prob)
    self.inter_layer = SketchIntermediate(hidden_dim, inter_dim, inter_activation)
    self.output = SketchOutput(inter_dim, hidden_dim, attention_norm_type, output_dropout_prob)
def __init__(self, num_heads, hidden_dim, inter_dim, attention_norm_type, inter_activation, attention_dropout_prob, hidden_dropout_prob, output_dropout_prob):
    super(SketchLayer, self).__init__()
    self.attention = SketchMultiHeadAttention(num_heads, hidden_dim, attention_norm_type, attention_dropout_prob, hidden_dropout_prob)
    self.inter_layer = SketchIntermediate(hidden_dim, inter_dim, inter_activation)
    self.output = SketchOutput(inter_dim, hidden_dim, attention_norm_type, output_dropout_prob)