我试图将两个矩阵列相关。即将第一个矩阵的第一列与第二个矩阵的第一列相关联，依此类推。在 numpy 我做：
np.corrcoef(x, y, axis=0)
而且效果很好。该命令的 Tensorflow 等效项是什么？
我尝试使用 streaming_pearson_correlation 1但这将所有列关联在一起而不是提供每列的结果。
作为最后的手段，我正在考虑将张量拆分为单独的列张量，但我猜这会产生性能成本。
我知道我可以将 numpy 包装在 py_func 中，但它不会在 GPU 上运行。
在此先感谢您的帮助。