与 Tensorflow 的自相关

Question

我想计算一些值与 tensorflow 的自相关。 我可以使用 scipy / numpy 进行计算，但我还没有弄清楚是否可以使用 tensorflow。

我想要的是：

import tensorflow as tf
from scipy import signal
import numpy as np
import tensorflow_probability as tfp
import matplotlib.pyplot as plt

test_data = tf.random.normal((100,))

plt.plot(signal.correlate(test_data, test_data, mode='full', method='auto'))
plt.plot(np.correlate(test_data, test_data, mode='full'))

正如预期的那样，scipy 和 numpy 的输出是相同的。 使用 TensorFlow 我试过了

plt.plot(tfp.stats.auto_correlation(test_data))

我最初假设会这样做，但会给出完全不同的结果。 有没有和 numpy / scipy 一样的 tensorflow 函数？

Answer 1

试试这个

td = tf.pad(test_data, [[0, len(test_data)]])[..., tf.newaxis]
plt.plot(tf.nn.conv1d(td[tf.newaxis, :], td[:, tf.newaxis], stride=1, padding='SAME')[0, :, 0])