gpt4 book ai didi

tensorflow - 错误 "Only integers, slices (` :`), ellipsis (` .. .`), tf.newaxis (` None`) 和标量”

转载 作者:行者123 更新时间:2023-12-03 23:10:21 24 4
gpt4 key购买 nike

我正在尝试使用 Tensorflow 后端在 Keras 中训练深度学习算法。我正在尝试执行以下操作:

x = tf.reshape(theta, [-1])[K.argmax(image)]

哪里 image是输入和 eta是一个坐标。我试图压平 theta,但出现错误
 Only integers, slices (`:`), ellipsis (`...`), tf.newaxis (`None`) and scalar tf.int32/tf.int64 tensors are valid indices, got <tf.Tensor 'loss_42/dense_264_loss/ArgMax:0' shape=(25,) dtype=int64>

最佳答案

我猜您想根据 K.argmax(image) 获得 theta 值.您不能在 tensorflow 中直接使用花哨的索引样式。 tf.gather可以实现这一点。

res = tf.gather(tf.reshape(theta, [-1]), K.argmax(image))

关于tensorflow - 错误 "Only integers, slices (` :`), ellipsis (` .. .`), tf.newaxis (` None`) 和标量”,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58575984/

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