Я предлагаю этот док.Но я не работаю с этим фреймворком.
Я установил его, используя pip install --upgrade tensorflow-probability
.
import tensorflow as tf
import tensorflow_probability as tfp
x = tf.random_uniform((5,2),2,3)
y = tf.random_uniform((5,2),2,3)
corr = tfp.stats.correlation(x, y)
with tf.Session() as sess :
print(sess.run( corr))
[[0.32789752 -0.12169117]
[0.83670807-0.09973542]]