TypeError: Got an unexpected keyword argument 'keep_dims'
时间: 2023-10-23 09:27:17 浏览: 311
Error Encountered an improper argument
这个错误通常出现在 TensorFlow 版本升级后,由于某些参数的变化导致代码出现错误。在 TensorFlow 1.x 中,`keep_dims` 是可以用作参数的,但在 TensorFlow 2.x 中,`keepdims` 取代了 `keep_dims`,以避免这种错误的发生。
要解决这个错误,你需要将 `keep_dims` 改为 `keepdims`。例如,如果你的代码中有这样的语句:
```
result = tf.reduce_sum(x, axis=1, keep_dims=True)
```
那么你需要将其改为:
```
result = tf.reduce_sum(x, axis=1, keepdims=True)
```
这样就能避免这个错误了。
阅读全文