python - TensorFlow 中的关联(如 numpy.correlate)

Question

TensorFlow 中是否有 CORRELATE 函数(如 numpy.correlate )？例如，

numpy.correlate(x,y)

Answer 1

我建议this文档。但我不使用这个框架。

我使用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]]