python - tf.keras.backend 替换小于 1 的张量值的方法

Question

我正在使用 Keras 和 Tensorflow 后端。

在我的损失函数中，我有一个张量，我需要将小于 1 的元素替换为 1。

我可以在文档中看到大量可用的函数 https://www.tensorflow.org/api_docs/python/tf/keras/backend

但我不知道该怎么做。

如果我这样做:

a_ = tf.Print(
    message='a_shape',
    input_=a_,
    data=[tf.shape(a_)]
)

我得到的形状为:

y_shape[128]

我本质上需要迭代这个张量，将小于 1 的元素替换为 1。

如何使用 keras tensorflow API 执行此操作？

谢谢-

Answer 1

如果a是你的张量，你可以执行以下操作:

b = a*tf.cast(a>1, 'float32') + tf.cast(a<=1, 'float32')