Calculating the cosine similarity of two tensors with pytorch

Question

I\'m new to pytorch and I\'m trying to implement the cosine similarity function. for some reason I get an error that suggests that my implementation only fits 1D vectors. Ho