如何在 Keras 中裁剪一个张量，每个子张量都有不同的裁剪值？

Posted 2023-03-23

【中文标题】如何在 Keras 中裁剪一个张量，每个子张量都有不同的裁剪值？

【英文标题】：How to clip a tensor in Keras, with a different clipping value for each subtensor?

【发布时间】：2020-04-12 12:14:18

【问题描述】：

我有一个 4d 张量 W（2d 卷积的内核）和一个 1d 数组/张量 H，指定每个过滤器的裁剪值。这意味着我有

len(H) == W.shape[-1]

并且我希望将 W[:,:,:,i] 的所有元素剪辑在 -H[i] 和 H[i] 之间。

我查看了keras.backend.clip，它说您可以将最小值和最大值指定为张量，但您不能指定轴。如果有人能告诉我如何实现这一目标，我将不胜感激。提前致谢。

【参考方案1】：

把它放在某个层内：

H = K.reshape(H, (1, 1, 1, -1))
topClip = K.minimum(H, W)
clipped = K.maximum(-H, topClip)

---

正如 Diego Palacios 所评论的，虽然 clip 接受张量，但它只接受具有 1 个元素的张量。

【讨论】：

谢谢！第一个选项工作正常，但第二个选项不行。 K.clip 似乎只接受 1 个元素的张量。