Id like to see what happens if you change the K,V matrix into a 3 dimensional tensor.