Using torch normalize for cosine similarity normalization

[ancestor-mithril]

This version is more clear and torch.nn.functional.normalize is faster in micro-benchmarks than the current handwritten implementation.