如何使用张量流计算相邻窗口中的均方误差？

问题描述

我将 tensorflow 与 keras 后端 (K) 一起使用，我使用 channel_first 实现

我想实现一个类似 MSE 的损失函数，但不是：

def mean_squared_error(y_true,y_pred):
    return K.mean(K.square(y_pred - y_true),axis=-1)

我希望该函数使用 5 个相邻像素，类似于以下代码段：

def mean_squared_error(y_true,y_pred):
    return K.mean(K.square(y_pred[-2:2] - y_true[-2:2]),axis=-1)

我该如何实施？

如何将相邻像素传递给损失函数？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）