I have one tensor x of shape [64,324]. Think of 64 as batch size and 324 as number of features. I want to perform cosine similarity with another tensor w such that that the output tensor is of size [64,10] , where 64 is same as batch size and 10 is output features. I could not find any tensor w which serves this purpose. w should be independent of batch size.