请问用Python做神经网络的时候有什么函数可以用来衡量输入参数和输出的相关性？

问题遇到的现象和发生背景

问题相关代码，请勿粘贴截图

运行结果及报错内容

我的解答思路和尝试过的方法

我想要达到的结果

论文中的相关性公式如上图。