calculate cosine similarity in Pytorch

Question

I have a candidate document embedding tensor, namely cdd_doc_embeddings of size [batch_size, cdd_size, signal_length, embedding_dim], a history cli