Correlation metrics optimization

[nicolaspi]

Optimizes the streaming correlation metrics by removing a while loop as well as usage of tf.sparse.add that computes only on CPU.

[bot-of-gabrieldemarmiesse]

@sorensenjs

You are owner of some files modified in this pull request. Would you kindly review the changes whenever you have the time to? Thank you very much.