gpt4 book ai didi

tensorflow - 将张量的所有负值设置为零(在 tensorflow 中)

转载 作者:行者123 更新时间:2023-12-03 09:44:45 25 4
gpt4 key购买 nike

这是我的问题。我有一个张量 X,我想将所有负值设置为零。在 numpy 中,我会执行以下操作 np.maximum(0, X) .有什么办法可以在 tensorflow 中达到同样的效果吗?我试过 tf.maximum(tf.fill(X.get_shape(), 0.0), X) ,但这会抛出 ValueError: Cannot convert a partially known TensorShape to a Tensor: (?,) .

附注。 X 是形状为 (?,) 的一维张量。

最佳答案

碰巧,您的问题与计算 rectifier activation function 完全相同。 ,并且 TensorFlow 有一个内置操作符, tf.nn.relu() ,这正是您所需要的:

X_with_negatives_set_to_zero = tf.nn.relu(X)

关于tensorflow - 将张量的所有负值设置为零(在 tensorflow 中),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41043894/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com