queries = inputs @ W_query
keys = inputs @ W_key
values = inputs @ W_value

        __
        __