直接上代码，原理大家都可以查到

注意：以下img自己加载即可，我是事先把图片转换成了单通道的Array形式，并且我的img是归一化后在0-1的。注意：以下img自己加载即可，我是事先把图片