output = self.scaled_dot_product_attention(queries, keys, values, valid_lens)