我尝试为Keras序列模型（特别是MNIST数据集）使用自定义平方根激活函数。当我使用tf.math.sqrt(x)时，训练进行得很顺利，模型也相当准确。但是，当我尝试使用tf.math.pow(x, 0.5)时，模型训练失败，损失变为NaN。
我真的不知道为什么会发生这种情况，因为我会认为这两种选择是相同的。
平方根函数

def tfsqrt(x):
    cond = tf.greater_equal(x, 0)
    return tf.where(cond, tf.math.sqrt(x), -tf.math.sqrt(-x))

幂函数

def pwsqrt(x):
  cond = tf.greater_equal(x, 0)
  return tf.where(cond, tf.math.pow(x, 0.5), -tf.math.pow(-x, 0.5))

如果有人能解释这种意外的行为，那将是非常感激的。谢谢！