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

Question

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

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

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

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

Answer 1

把它放在某个层内：

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

---

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