与tensorlflow中的每个特征具有高效的多特征相似性

I want to calculate multi-feature similarity with each feature in tensorflow. But I don't know how to write it efficiently. Here is my sample code:I want to calculate multi-feature similarity wi